Update app.py

app.py

@@ -16,7 +16,7 @@ def detect_objects(image):
         "yolo",  # YOLOv8 CLI command
         "task=detect",  # Specify the task as detection
         "mode=predict",  # Set mode to predict
-        f"model=best_p6.pt",  # Path to your YOLOv8 model weights
+        f"model=best.pt",  # Path to your YOLOv8 model weights
         f"source={input_image_path}",  # Input image path (uploaded by user)
         f"project={output_dir}",  # Output directory
         f"name=result",  # Subfolder name for results
@@ -42,7 +42,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="file"),  # Return an image as output
+    outputs=gr.Image(type="filepath"),  # Return an image file path as output
     title="YOLOv8 Object Detection",
     description="Upload an image of floating waste in water, and this app will detect it using YOLOv8."
 )