Update app.py

app.py

@@ -18,6 +18,9 @@ def model():
     
     inputs = image_processor(images=image, return_tensors="pt")
     outputs = model(**inputs)
+
+    # display image
+    st.image('http://images.cocodataset.org/val2017/000000039769.jpg', caption='Sunrise by the mountains')
     
     # convert outputs (bounding boxes and class logits) to Pascal VOC format (xmin, ymin, xmax, ymax)
     target_sizes = torch.tensor([image.size[::-1]])