akhaliq HF staff commited on
Commit
f859d50
1 Parent(s): 5bd6fbb
Files changed (2) hide show
  1. app.py +2 -2
  2. requirements.txt +2 -1
app.py CHANGED
@@ -5,9 +5,9 @@ import gradio as gr
5
  MODEL_DIR = 'models/pretrain'
6
  os.makedirs(MODEL_DIR, exist_ok=True)
7
 
8
- os.system("wget https://hkustconnect-my.sharepoint.com/:u:/g/personal/jzhubt_connect_ust_hk/ETYVen9KXGlAia2gH6pcZswB9Lw-21vWrE75OACvG2SBow\?e\=SCGqg0\&download=1 -O $MODEL_DIR/stylegan2-ffhq-config-f-1024x1024.pth --quiet")
9
-
10
 
 
11
  # python 3.7
12
  """Demo."""
13
  import io
 
5
  MODEL_DIR = 'models/pretrain'
6
  os.makedirs(MODEL_DIR, exist_ok=True)
7
 
8
+ os.system("gdown https://drive.google.com/uc?id=1--h-4E5LSxe6VTp9rjJhtoILxxNH0X5n")
 
9
 
10
+ os.system("cp stylegan2-ffhq-config-f-1024x1024.pth models/pretrain/stylegan2-ffhq-config-f-1024x1024.pth")
11
  # python 3.7
12
  """Demo."""
13
  import io
requirements.txt CHANGED
@@ -8,4 +8,5 @@ requests
8
  bs4
9
  tqdm
10
  rich
11
- ninja
 
 
8
  bs4
9
  tqdm
10
  rich
11
+ ninja
12
+ gdown