Update app.py

app.py

@@ -2,21 +2,19 @@ import gradio as gr
 from ultralytics import YOLO
 import os
 
-# Load the YOLO model
+# Download the model from Google Drive if it's not present
 model_path = "best.pt"
+
+# Load the YOLOv8 model
 model = YOLO(model_path)
 
 # Define the object detection function
 def detect_objects(image):
     results = model(image)
-
-    # Extract logs from YOLO results
-    logs = results[0].verbose()
-
-    return results[0].plot(), logs  # Return image + logs
+    return results[0].plot()
 
 # Create a Gradio interface
-app = gr.Interface(fn=detect_objects, inputs="image", outputs=["image", "text"])
+app = gr.Interface(fn=detect_objects, inputs="image", outputs="image")
 
 # Launch Gradio app
-app.launch()
+app.launch()