Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import subprocess
 import os
+from PIL import Image
 
 # YOLOv8 detection function
 def detect_objects(image):
@@ -32,9 +33,9 @@ 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
+    # Return the output image for display as a PIL image
     if os.path.exists(output_image_path):
-        return output_image_path
+        return Image.open(output_image_path)
     else:
         return "Error: Output image not found."
 
@@ -42,7 +43,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="filepath"),  # Return an image file path as output
+    outputs=gr.Image(type="pil"),  # Return a PIL image for display
     title="YOLOv8 Object Detection",
     description="Upload an image of floating waste in water, and this app will detect it using YOLOv8."
 )