add gdown
Browse files- app.py +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("
|
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
|