Update app.py

app.py

@@ -29,6 +29,8 @@ def make_prediction(img, feature_extractor, model):
 def detect_objects(model_name, image_input, threshold, display_mask=False):
     feature_extractor = AutoFeatureExtractor.from_pretrained(MODELS_REPO[model_name])
 
+    print(image_input)
+
     if "DETR" in model_name:
         model = DetrForObjectDetection.from_pretrained(MODELS_REPO[model_name])
         model_details = "DETR details"
@@ -43,7 +45,7 @@ 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.name)
+        image_path = pathlib.Path(image_input)
         mask_path = image_path.parent / (
             image_path.stem.replace("_HE", "_mask") + image_path.suffix
         )