Spaces:

RamAnanth1
/

ControlNet

Runtime error

RamAnanth1 commited on Feb 11, 2023

Commit

4963e4b

•

1 Parent(s): ef43b24

Update annotator/openpose/init.py

Files changed (1) hide show

annotator/openpose/__init__.py CHANGED Viewed

@@ -7,8 +7,10 @@ from . import util
 from .body import Body
 from .hand import Hand
-body_estimation = Body('./annotator/ckpts/body_pose_model.pth')
-hand_estimation = Hand('./annotator/ckpts/hand_pose_model.pth')
 def apply_openpose(oriImg, hand=False):

 from .body import Body
 from .hand import Hand
+from huggingface_hub import hf_hub_url, hf_hub_download
+REPO_ID = "lllyasviel/ControlNet"
+body_estimation = Body(hf_hub_download(REPO_ID+'/annotator/ckpts/body_pose_model.pth'))
+hand_estimation = Hand(hf_hub_download(REPO_ID+'/annotator/ckpts/hand_pose_model.pth'))
 def apply_openpose(oriImg, hand=False):