Create app.py

app.py

@@ -0,0 +1,26 @@
+import cv2
+import gradio as gr
+from ultralytics import YOLO
+
+# Load YOLOv8-Pose model (downloads if not cached)
+model = YOLO("yolov8s-pose.pt")  # You can also use yolov8s/m/l/x-pose.pt
+
+def predict_pose(frame):
+    # Run YOLOv8-Pose inference
+    results = model.predict(frame, imgsz=640, conf=0.5)[0]
+
+    # Draw results on frame
+    annotated_frame = results.plot()  # YOLOv8 handles drawing
+    return annotated_frame
+
+# Set up Gradio interface
+with gr.Blocks() as demo:
+    with gr.Row():
+        with gr.Column():
+            input_img = gr.Image(sources=["webcam"])
+        with gr.Column():
+            output_img = gr.Image(streaming=True)
+        input_img.stream(predict_pose, input_img, output_img,
+                                time_limit=30, stream_every=0.1, concurrency_limit=30)
+
+demo.launch()