move location
Browse files
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('
|
|
|
|
| 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 = "
|
| 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):
|