Spaces:
Runtime error
Runtime error
Fixed model downloads
Browse files
app.py
CHANGED
@@ -17,6 +17,7 @@ import tempfile
|
|
17 |
import shutil
|
18 |
from argparse import Namespace
|
19 |
from pathlib import Path
|
|
|
20 |
|
21 |
import dlib
|
22 |
import numpy as np
|
@@ -32,13 +33,19 @@ os.makedirs(model_dir, exist_ok=True)
|
|
32 |
|
33 |
models_and_paths = {"akhaliq/JoJoGAN_e4e_ffhq_encode": "e4e_ffhq_encode.pt",
|
34 |
"akhaliq/jojogan_dlib": "shape_predictor_68_face_landmarks.dat",
|
35 |
-
"akhaliq/jojogan-stylegan2-ffhq-config-f": f
|
36 |
|
37 |
def get_models():
|
|
|
|
|
38 |
for repo_id, file_path in models_and_paths.items():
|
39 |
hf_hub_download(repo_id=repo_id, filename=file_path)
|
|
|
|
|
|
|
|
|
40 |
|
41 |
-
model_list = [
|
42 |
|
43 |
return model_list
|
44 |
|
|
|
17 |
import shutil
|
18 |
from argparse import Namespace
|
19 |
from pathlib import Path
|
20 |
+
import shutil
|
21 |
|
22 |
import dlib
|
23 |
import numpy as np
|
|
|
33 |
|
34 |
models_and_paths = {"akhaliq/JoJoGAN_e4e_ffhq_encode": "e4e_ffhq_encode.pt",
|
35 |
"akhaliq/jojogan_dlib": "shape_predictor_68_face_landmarks.dat",
|
36 |
+
"akhaliq/jojogan-stylegan2-ffhq-config-f": "stylegan2-ffhq-config-f.pt"}
|
37 |
|
38 |
def get_models():
|
39 |
+
os.makedirs(model_dir, exist_ok=True)
|
40 |
+
|
41 |
for repo_id, file_path in models_and_paths.items():
|
42 |
hf_hub_download(repo_id=repo_id, filename=file_path)
|
43 |
+
if not "akhaliq" in repo_id:
|
44 |
+
shutil.move(file_path, os.path.join(model_dir, file_path))
|
45 |
+
elif "stylegan2" in file_path:
|
46 |
+
shutil.move(file_path, os.path.join(model_dir, "base.pt"))
|
47 |
|
48 |
+
model_list = [Path(model_ckpt).stem for model_ckpt in os.listdir(model_dir)]
|
49 |
|
50 |
return model_list
|
51 |
|