Spaces:

linoyts
/

video-ViTPose-extractor

Paused

linoyts HF Staff commited on Feb 18

Commit

008509c

verified ·

1 Parent(s): ed65396

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -103,8 +103,8 @@ vitpose_simple = VitPoseForPoseEstimation.from_pretrained("usyd-community/vitpos
 vitpose_simple_processor = AutoProcessor.from_pretrained("usyd-community/vitpose-base-simple")
 # WholeBody ViTPose (133 keypoints)
-vitpose_wholebody = VitPoseForPoseEstimation.from_pretrained("usyd-community/vitpose-plus-huge-wholebody").to(device)
-vitpose_wholebody_processor = AutoProcessor.from_pretrained("usyd-community/vitpose-plus-huge-wholebody")
 print("All models loaded!")

 vitpose_simple_processor = AutoProcessor.from_pretrained("usyd-community/vitpose-base-simple")
 # WholeBody ViTPose (133 keypoints)
+vitpose_wholebody = VitPoseForPoseEstimation.from_pretrained("usyd-community/vitpose-plus-huge").to(device)
+vitpose_wholebody_processor = AutoProcessor.from_pretrained("usyd-community/vitpose-plus-huge")
 print("All models loaded!")