Hugging Face's logo

Spaces:
RamAnanth1
/
ControlNet
Runtime error

App Files Community
12
RamAnanth1 commited on
Commit
4963e4b
·
1 Parent(s): ef43b24

Update annotator/openpose/__init__.py

Browse files
Files changed (1) hide show
  1. annotator/openpose/__init__.py +4 -2
annotator/openpose/__init__.py CHANGED
@@ -7,8 +7,10 @@ from . import util
7
  from .body import Body
8
  from .hand import Hand
9
 
10
- body_estimation = Body('./annotator/ckpts/body_pose_model.pth')
11
- hand_estimation = Hand('./annotator/ckpts/hand_pose_model.pth')
 
 
12
 
13
 
14
  def apply_openpose(oriImg, hand=False):
 
7
  from .body import Body
8
  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):