radames commited on
Commit
baf08a6
1 Parent(s): f91d80b

move location

Browse files
Files changed (1) hide show
  1. visualizer_drag_gradio.py +3 -2
visualizer_drag_gradio.py CHANGED
@@ -22,7 +22,8 @@ from viz.renderer import Renderer, add_watermark_np
22
  from huggingface_hub import snapshot_download
23
 
24
  model_dir = Path('./checkpoints')
25
- snapshot_download('radames/DragGan', repo_type='model', local_dir=model_dir)
 
26
 
27
  parser = ArgumentParser()
28
  parser.add_argument('--share', action='store_true')
@@ -32,7 +33,7 @@ args = parser.parse_args()
32
  cache_dir = args.cache_dir
33
 
34
  device = 'cuda'
35
- IS_SPACE = "radames/DragGan" in os.environ.get('SPACE_ID', '')
36
 
37
 
38
  def reverse_point_pairs(points):
 
22
  from huggingface_hub import snapshot_download
23
 
24
  model_dir = Path('./checkpoints')
25
+ snapshot_download('DragGan/DragGan-Models',
26
+ repo_type='model', local_dir=model_dir)
27
 
28
  parser = ArgumentParser()
29
  parser.add_argument('--share', action='store_true')
 
33
  cache_dir = args.cache_dir
34
 
35
  device = 'cuda'
36
+ IS_SPACE = "DragGan/DragGan" in os.environ.get('SPACE_ID', '')
37
 
38
 
39
  def reverse_point_pairs(points):