Spaces:

RamAnanth1
/

ControlNet

Runtime error

RamAnanth1 commited on Feb 11, 2023

Commit

5f7ee3e

1 Parent(s): 32a888f

Update annotator/openpose/init.py

Files changed (1) hide show

annotator/openpose/__init__.py CHANGED Viewed

@@ -7,10 +7,10 @@ from . import util
 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(hf_hub_url(REPO_ID, '/annotator/ckpts/body_pose_model.pth')))
-hand_estimation = Hand(hf_hub_download(hf_hub_url(REPO_ID,'/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, cached_download
 REPO_ID = "lllyasviel/ControlNet"
+body_estimation = Body(cached_download(hf_hub_url(REPO_ID, 'annotator/ckpts/body_pose_model.pth')))
+hand_estimation = Hand(cached_download(hf_hub_url(REPO_ID,'annotator/ckpts/hand_pose_model.pth')))
 def apply_openpose(oriImg, hand=False):