Update app.py

app.py

@@ -1,29 +1,21 @@
 import gradio as gr
 from ultralytics import YOLO
 import os
-import sys
-import io
 
-# Load the YOLOv8 model
-model = YOLO("best.pt")
+# Load the YOLO model
+model_path = "best.pt"
+model = YOLO(model_path)
 
 # Define the object detection function
 def detect_objects(image):
-    # Capture logs
-    old_stdout = sys.stdout
-    sys.stdout = io.StringIO()
-
-    # Run YOLO inference
     results = model(image)
 
-    # Get inference logs
-    logs = sys.stdout.getvalue()
-    sys.stdout = old_stdout  # Restore stdout
+    # Extract logs from YOLO results
+    logs = results[0].verbose()
 
-    # Return both the image with detections and logs as text
-    return results[0].plot(), logs
+    return results[0].plot(), logs  # Return image + logs
 
-# Create a Gradio interface with two outputs (image + logs)
+# Create a Gradio interface
 app = gr.Interface(fn=detect_objects, inputs="image", outputs=["image", "text"])
 
 # Launch Gradio app