Update app.py

app.py

@@ -191,11 +191,12 @@ def detections_to_json(detections):
         detections_list.append(detection_dict)
     return detections_list
 
-def process_images(images):
+def process_images(image_files):
     labels = ["insect"]
     results = []
     json_outputs = []
-    for image in images:
+    for image_file in image_files:
+        image = Image.open(image_file).convert("RGB")
         original_image, detections = grounded_segmentation(image, labels, threshold=0.3, polygon_refinement=True)
         yellow_background_with_insects = create_yellow_background_with_insects(np.array(original_image), detections)
         detections_json = detections_to_json(detections)
@@ -208,7 +209,7 @@ def process_images(images):
 
 gr.Interface(
     fn=process_images,
-    inputs=gr.File(file_count="multiple", type="file"),
+    inputs=gr.File(file_count="multiple", type="filepath"),
     outputs=[gr.Gallery(), gr.Textbox()],
     title="🐞 InsectSAM + GroundingDINO Inference",
 ).launch()