Update app.py

app.py

@@ -29,7 +29,7 @@ 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.filename)
+    print(image_input.value)
 
     if "DETR" in model_name:
         model = DetrForObjectDetection.from_pretrained(MODELS_REPO[model_name])
@@ -45,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.filename)
+        image_path = pathlib.Path(image_input.value)
         mask_path = image_path.parent / (
             image_path.stem.replace("_HE", "_mask") + image_path.suffix
         )