Spaces:

Deci
/

YOLO-NAS-Pose-Demo

Running

harpreetsahota commited on Nov 6, 2023

Commit

99a0703

•

1 Parent(s): 202d2bb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ from super_gradients.training.utils.visualization.pose_estimation import PoseVis
 # Initialize your pose estimation model
 yolo_nas_pose = models.get("yolo_nas_pose_l",
                            num_classes=17,
-                           checkpoint_path="/content/yolo_nas_pose_l_coco_pose.pth")
 def process_and_predict(url=None,
                         image=None,

 # Initialize your pose estimation model
 yolo_nas_pose = models.get("yolo_nas_pose_l",
                            num_classes=17,
+                           checkpoint_path="./yolo_nas_pose_l_coco_pose.pth")
 def process_and_predict(url=None,
                         image=None,