Update app.py

app.py

@@ -1,20 +1,30 @@
-import gradio as gr
-from ultralytics import YOLO
-import os
-
-# 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)
-    return results[0].plot()
-
-# Create a Gradio interface
-app = gr.Interface(fn=detect_objects, inputs="image", outputs="image")
-
-# Launch Gradio app
-app.launch()
+import gradio as gr
+from ultralytics import YOLO
+import os
+import sys
+import io
+
+# Load the YOLOv8 model
+model = YOLO("best.pt")
+
+# 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
+
+    # Return both the image with detections and logs as text
+    return results[0].plot(), logs
+
+# Create a Gradio interface with two outputs (image + logs)
+app = gr.Interface(fn=detect_objects, inputs="image", outputs=["image", "text"])
+
+# Launch Gradio app
+app.launch()