Added example images

app.py

@@ -51,7 +51,8 @@ class ObjectDetection:
         self.model.load_state_dict(model_ckpt)
         self.model.eval()
 
-    def predict(self, img, score_threshold, iou_threshold):
+    def predict(self, img):
+        score_threshold, iou_threshold = 0.05, 0.1
         img_w, img_h = img.size
         inp = self.test_transform(image=np.array(img.convert("RGB")))["image"]
         out = self.model(inp.unsqueeze(0))
@@ -138,25 +139,15 @@ detector = ObjectDetection(ckpt_path=model_path)
 iface = gr.Interface(
     fn=detector.predict,
     inputs=[
-        gr.Image(type="pil", label="Input"),
-        gr.Slider(
-            minimum=0,
-            maximum=1,
-            step=0.05,
-            value=0.05,
-            label="Score Threshold",
-        ),
-        gr.Slider(
-            minimum=0,
-            maximum=1,
-            step=0.05,
-            value=0.1,
-            label="IoU Threshold",
+        gr.Image(
+            type="pil",
+            label="Input",
+            height=400,
         ),
     ],
     outputs=gr.AnnotatedImage(
-        height=600,
-        width=800,
+        height=400,
     ),
+    examples="Examples",
 )
 iface.launch()