Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -7,7 +7,7 @@ model = YOLO("yolo11n-pose.pt") # You can also use yolo11s/m/l/x-pose.pt
|
|
7 |
|
8 |
def predict_pose(frame):
|
9 |
# Run YOLO11-Pose inference
|
10 |
-
results = model.predict(frame, imgsz=
|
11 |
|
12 |
# Draw results on frame
|
13 |
annotated_frame = results.plot(labels=False, boxes=False) # YOLO11 handles drawing
|
|
|
7 |
|
8 |
def predict_pose(frame):
|
9 |
# Run YOLO11-Pose inference
|
10 |
+
results = model.predict(frame, imgsz=480, conf=0.5, iou=0.5)[0]
|
11 |
|
12 |
# Draw results on frame
|
13 |
annotated_frame = results.plot(labels=False, boxes=False) # YOLO11 handles drawing
|