Update app.py

app.py

@@ -33,9 +33,10 @@ def detect_objects(image):
     # Get the path to the output image
     output_image_path = os.path.join(output_dir, "result", os.path.basename(input_image_path))
 
-    # Return the output image for display as a PIL image
+    # Check if the image exists
     if os.path.exists(output_image_path):
-        return Image.open(output_image_path)
+        # Return the image directly to Gradio
+        return output_image_path  # Return file path for Gradio to display
     else:
         return "Error: Output image not found."
 
@@ -43,7 +44,7 @@ def detect_objects(image):
 interface = gr.Interface(
     fn=detect_objects,  # The function to call when an image is uploaded
     inputs=gr.Image(type="pil"),  # Accept images as input
-    outputs=gr.Image(type="pil"),  # Return a PIL image for display
+    outputs=gr.Image(type="filepath"),  # Return image file path for display
     title="YOLOv8 Object Detection",
     description="Upload an image of floating waste in water, and this app will detect it using YOLOv8."
 )