Spaces:

rinong
/

StyleGAN-NADA

Runtime error

rinong commited on May 16, 2022

Commit

1408f30

1 Parent(s): a390b6f

Added joker, anime models.

Files changed (1) hide show

app.py CHANGED Viewed

@@ -33,7 +33,10 @@ os.makedirs(model_dir, exist_ok=True)
 model_repos = {"e4e": ("akhaliq/JoJoGAN_e4e_ffhq_encode", "e4e_ffhq_encode.pt"),
                "dlib": ("akhaliq/jojogan_dlib", "shape_predictor_68_face_landmarks.dat"),
-               "base": ("akhaliq/jojogan-stylegan2-ffhq-config-f", "stylegan2-ffhq-config-f.pt")}
 def get_models():
     os.makedirs(model_dir, exist_ok=True)

 model_repos = {"e4e": ("akhaliq/JoJoGAN_e4e_ffhq_encode", "e4e_ffhq_encode.pt"),
                "dlib": ("akhaliq/jojogan_dlib", "shape_predictor_68_face_landmarks.dat"),
+               "base": ("akhaliq/jojogan-stylegan2-ffhq-config-f", "stylegan2-ffhq-config-f.pt"),
+               "anime": ("rinong/stylegan-nada-models", "anime.pt"),
+               "joker": ("rinong/stylegan-nada-models", "joker.pt")
+               }
 def get_models():
     os.makedirs(model_dir, exist_ok=True)