Update app.py

app.py

@@ -62,7 +62,13 @@ def inference(image, min_score, model_name):
     models[model_id]["cfg"].MODEL.ROI_HEADS.SCORE_THRESH_TEST = min_score
     predictor = DefaultPredictor(models[model_id]["cfg"])
 
-    for img in image:
+    # Asegurémonos de que image es una lista de imágenes o solo una
+    if isinstance(image, list):  # Si es una lista de imágenes
+        images = image
+    else:  # Si es solo una imagen, la convertimos en una lista
+        images = [image]
+
+    for img in images:
         # Convert image from BGR to RGB if necessary
         im = img[:,:,::-1]
 
@@ -71,13 +77,13 @@ def inference(image, min_score, model_name):
 
         v = Visualizer(im, models[model_id]["metadata"], scale=1.2)
         out = v.draw_instance_predictions(outputs["instances"].to("cpu"))
-        
+
         # Convert the result from BGR to RGB
         result_image = out.get_image()
         result_image_rgb = result_image[:, :, ::-1]  # Convert BGR to RGB
 
         results.append(result_image_rgb)  # Add the processed image to the list
-    
+
     return results  # Return all the results
 
 title = "# DBMDZ Detectron2 Model Demo"
@@ -100,7 +106,7 @@ with gr.Blocks() as demo:
 
     model_name = gr.Radio(choices=[model["name"] for model in models], value=models[0]["name"], label="Select Detectron2 model")
 
-    output_gallery = gr.Gallery(label="Output Images", elem_id="output_images")
+    output_gallery = gr.Gallery(type ="numpy", label="Output Images", elem_id="output_images")
 
     inference_button = gr.Button("Submit")