SUPERSHANKY commited on
Commit
6811c76
·
1 Parent(s): 38ef057

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -13,7 +13,7 @@ if os.getenv('SYSTEM') == 'spaces':
13
  with open('patch') as f:
14
  subprocess.run(shlex.split('patch -p1'), stdin=f, cwd='ControlNet')
15
 
16
- base_url = 'https://huggingface.co/lllyasviel/ControlNet/tree/main/annotator/ckpts'
17
  names = [
18
  'body_pose_model.pth',
19
  'dpt_hybrid-midas-501f0c75.pt',
@@ -24,7 +24,7 @@ names = [
24
  'upernet_global_small.pth',
25
  ]
26
  for name in names:
27
- command = f'wget https://huggingface.co/lllyasviel/ControlNet/tree/main/annotator/ckpts/{name} -O {name}'
28
  out_path = pathlib.Path(f'ControlNet/annotator/ckpts/{name}')
29
  if out_path.exists():
30
  continue
 
13
  with open('patch') as f:
14
  subprocess.run(shlex.split('patch -p1'), stdin=f, cwd='ControlNet')
15
 
16
+ base_url = 'https://huggingface.co/lllyasviel/ControlNet/tree/resolve/annotator/ckpts'
17
  names = [
18
  'body_pose_model.pth',
19
  'dpt_hybrid-midas-501f0c75.pt',
 
24
  'upernet_global_small.pth',
25
  ]
26
  for name in names:
27
+ command = f'wget https://huggingface.co/lllyasviel/ControlNet/tree/resolve/annotator/ckpts/{name} -O {name}'
28
  out_path = pathlib.Path(f'ControlNet/annotator/ckpts/{name}')
29
  if out_path.exists():
30
  continue