subh71 commited on
Commit
522da43
·
verified ·
1 Parent(s): d86f0cd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -20,7 +20,7 @@ def model():
20
  outputs = model(**inputs)
21
 
22
  # display image
23
- st.image('http://images.cocodataset.org/val2017/000000039769.jpg', caption='Sunrise by the mountains')
24
 
25
  # convert outputs (bounding boxes and class logits) to Pascal VOC format (xmin, ymin, xmax, ymax)
26
  target_sizes = torch.tensor([image.size[::-1]])
 
20
  outputs = model(**inputs)
21
 
22
  # display image
23
+ st.image(image, caption='Animal Images')
24
 
25
  # convert outputs (bounding boxes and class logits) to Pascal VOC format (xmin, ymin, xmax, ymax)
26
  target_sizes = torch.tensor([image.size[::-1]])