Update app.py

app.py

@@ -26,11 +26,9 @@ def make_prediction(img, feature_extractor, model):
     )
 
 
-def detect_objects(model_name, image_input, threshold, display_mask=False):
+def detect_objects(model_name, image_input, image_path, threshold, display_mask=False):
     feature_extractor = AutoFeatureExtractor.from_pretrained(MODELS_REPO[model_name])
 
-    print(image_input.value)
-
     if "DETR" in model_name:
         model = DetrForObjectDetection.from_pretrained(MODELS_REPO[model_name])
         model_details = "DETR details"
@@ -44,8 +42,6 @@ def detect_objects(model_name, image_input, threshold, display_mask=False):
 
     mask_pil_image = None
     if display_mask:
-        # get image path
-        image_path = pathlib.Path(image_input.value)
         mask_path = image_path.parent / (
             image_path.stem.replace("_HE", "_mask") + image_path.suffix
         )
@@ -71,11 +67,9 @@ def detect_objects(model_name, image_input, threshold, display_mask=False):
     )
 
 
-def set_example_image(example: list) -> dict:
-    print(example)
-    print(example[0])
-    print(gr.Image.update(value=example[0]))
-    return gr.Image.update(value=example[0])
+def set_example_image(example: list):
+    img_path = example[0]
+    return gr.Image.update(value=example[0]), img_path
 
 
 with gr.Blocks(css=css) as app:
@@ -136,7 +130,7 @@ with gr.Blocks(css=css) as app:
 
     detect_button.click(
         detect_objects,
-        inputs=[options, img_input, slider_input, display_mask],
+        inputs=[options, img_input, img_path, slider_input, display_mask],
         outputs=[
             img_output_from_upload,
             decoder_att_map_output,
@@ -147,8 +141,7 @@ with gr.Blocks(css=css) as app:
         queue=True,
     )
     example_images.click(
-        fn=set_example_image, inputs=[example_images], outputs=[img_input]
+        fn=set_example_image, inputs=[example_images], outputs=[img_input, img_path]
     )
 
     app.launch(enable_queue=True)
-