Create app.py

app.py

@@ -0,0 +1,25 @@
+import cv2
+from ultralytics import YOLO
+import gradio as gr
+
+# Load the YOLOv8 model
+model = YOLO('best.pt')
+
+# Define the Gradio interface
+def yolo_tracking(frame):
+    # Run YOLOv8 tracking on the frame
+    results = model.track(frame, persist=True)
+
+    # Visualize the results on the frame
+    annotated_frame = results[0].plot()
+
+    # Convert the frame to RGB for Gradio display
+    rgb_frame = cv2.cvtColor(annotated_frame, cv2.COLOR_BGR2RGB)
+
+    return rgb_frame
+
+# Create the Gradio interface
+interface = gr.Interface(fn=yolo_tracking, inputs=gr.Video(source="webcam"), outputs=gr.Image(title="YOLOv8 Tracking"))
+
+# Launch the Gradio interface
+interface.launch()