Spaces:

hylee
/

UGATIT

Sleeping

App Files Files Community

hylee commited on Apr 16, 2022

Commit

a6944f9

1 Parent(s): a8ceee9

init

Browse files

Files changed (1) hide show

app.py +10 -9

app.py CHANGED Viewed

@@ -45,28 +45,29 @@ def parse_args() -> argparse.Namespace:
 def load_checkpoint():
-    dir = 'UGATIT_selfie2anime_lsgan_4resblock_6dis_1_1_10_10_1000_sn_smoothing'
     checkpoint_path = huggingface_hub.hf_hub_download(MODEL_REPO,
-                                                dir+'/checkpoint',
-                                                cache_dir=dir,
                                                 force_filename='checkpoint')
     print(checkpoint_path)
     checkpoint_path = huggingface_hub.hf_hub_download(MODEL_REPO,
-                                                      dir + '/UGATIT.model-1000000.index',
-                                                      cache_dir=dir,
                                                       force_filename='UGATIT.model-1000000.index')
     print(checkpoint_path)
     checkpoint_path = huggingface_hub.hf_hub_download(MODEL_REPO,
-                                                      dir + '/UGATIT.model-1000000.meta',
-                                                      cache_dir=dir,
                                                       force_filename='UGATIT.model-1000000.meta')
     print(checkpoint_path)
     checkpoint_path = huggingface_hub.hf_hub_download(MODEL_REPO,
-                                                      dir + '/UGATIT.model-1000000.data-00000-of-00001',
-                                                      cache_dir=dir,
                                                       force_filename='UGATIT.model-1000000.data-00000-of-00001')
     print(checkpoint_path)

 def load_checkpoint():
+    dir = 'checkpoint'
+    name = 'UGATIT_selfie2anime_lsgan_4resblock_6dis_1_1_10_10_1000_sn_smoothing'
     checkpoint_path = huggingface_hub.hf_hub_download(MODEL_REPO,
+                                                name+'/checkpoint',
+                                                cache_dir=dir+'/'+name,
                                                 force_filename='checkpoint')
     print(checkpoint_path)
     checkpoint_path = huggingface_hub.hf_hub_download(MODEL_REPO,
+                                                      name + '/UGATIT.model-1000000.index',
+                                                      cache_dir=dir+'/'+name,
                                                       force_filename='UGATIT.model-1000000.index')
     print(checkpoint_path)
     checkpoint_path = huggingface_hub.hf_hub_download(MODEL_REPO,
+                                                      name + '/UGATIT.model-1000000.meta',
+                                                      cache_dir=dir+'/'+name,
                                                       force_filename='UGATIT.model-1000000.meta')
     print(checkpoint_path)
     checkpoint_path = huggingface_hub.hf_hub_download(MODEL_REPO,
+                                                      name + '/UGATIT.model-1000000.data-00000-of-00001',
+                                                      cache_dir=dir+'/'+name,
                                                       force_filename='UGATIT.model-1000000.data-00000-of-00001')
     print(checkpoint_path)