hylee commited on
Commit
a8ceee9
·
1 Parent(s): 1b9c487
Files changed (1) hide show
  1. app.py +24 -3
app.py CHANGED
@@ -45,11 +45,32 @@ def parse_args() -> argparse.Namespace:
45
 
46
 
47
  def load_checkpoint():
 
48
  checkpoint_path = huggingface_hub.hf_hub_download(MODEL_REPO,
49
- 'UGATIT_selfie2anime_lsgan_4resblock_6dis_1_1_10_10_1000_sn_smoothing/checkpoint',
50
- cache_dir='UGATIT_selfie2anime_lsgan_4resblock_6dis_1_1_10_10_1000_sn_smoothing')
 
51
  print(checkpoint_path)
52
- return 'UGATIT_selfie2anime_lsgan_4resblock_6dis_1_1_10_10_1000_sn_smoothing'
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
53
 
54
 
55
  def run(
 
45
 
46
 
47
  def load_checkpoint():
48
+ dir = 'UGATIT_selfie2anime_lsgan_4resblock_6dis_1_1_10_10_1000_sn_smoothing'
49
  checkpoint_path = huggingface_hub.hf_hub_download(MODEL_REPO,
50
+ dir+'/checkpoint',
51
+ cache_dir=dir,
52
+ force_filename='checkpoint')
53
  print(checkpoint_path)
54
+
55
+ checkpoint_path = huggingface_hub.hf_hub_download(MODEL_REPO,
56
+ dir + '/UGATIT.model-1000000.index',
57
+ cache_dir=dir,
58
+ force_filename='UGATIT.model-1000000.index')
59
+ print(checkpoint_path)
60
+
61
+ checkpoint_path = huggingface_hub.hf_hub_download(MODEL_REPO,
62
+ dir + '/UGATIT.model-1000000.meta',
63
+ cache_dir=dir,
64
+ force_filename='UGATIT.model-1000000.meta')
65
+ print(checkpoint_path)
66
+
67
+ checkpoint_path = huggingface_hub.hf_hub_download(MODEL_REPO,
68
+ dir + '/UGATIT.model-1000000.data-00000-of-00001',
69
+ cache_dir=dir,
70
+ force_filename='UGATIT.model-1000000.data-00000-of-00001')
71
+ print(checkpoint_path)
72
+
73
+ return dir
74
 
75
 
76
  def run(