Hugging Face's logo

Spaces:
RamAnanth1
/
ControlNet
Runtime error

App Files Community
12
RamAnanth1 commited on
Commit
15a3d8e
•
1 Parent(s): 4963e4b

Update annotator/openpose/__init__.py

Browse files
Files changed (1) hide show
  1. annotator/openpose/__init__.py +3 -2
annotator/openpose/__init__.py CHANGED
@@ -9,8 +9,9 @@ from .hand import Hand
9
 
10
  from huggingface_hub import hf_hub_url, hf_hub_download
11
  REPO_ID = "lllyasviel/ControlNet"
12
- body_estimation = Body(hf_hub_download(REPO_ID+'/annotator/ckpts/body_pose_model.pth'))
13
- hand_estimation = Hand(hf_hub_download(REPO_ID+'/annotator/ckpts/hand_pose_model.pth'))
 
14
 
15
 
16
  def apply_openpose(oriImg, hand=False):
 
9
 
10
  from huggingface_hub import hf_hub_url, hf_hub_download
11
  REPO_ID = "lllyasviel/ControlNet"
12
+ hf_hub_download(hf_hub_url(REPO_ID, FILENAME))
13
+ body_estimation = Body(hf_hub_download(hf_hub_url(REPO_ID, '/annotator/ckpts/body_pose_model.pth')))
14
+ hand_estimation = Hand(hf_hub_download(hf_hub_url(REPO_ID,'/annotator/ckpts/hand_pose_model.pth')))
15
 
16
 
17
  def apply_openpose(oriImg, hand=False):