rinong commited on
Commit
1408f30
1 Parent(s): a390b6f

Added joker, anime models.

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -33,7 +33,10 @@ os.makedirs(model_dir, exist_ok=True)
33
 
34
  model_repos = {"e4e": ("akhaliq/JoJoGAN_e4e_ffhq_encode", "e4e_ffhq_encode.pt"),
35
  "dlib": ("akhaliq/jojogan_dlib", "shape_predictor_68_face_landmarks.dat"),
36
- "base": ("akhaliq/jojogan-stylegan2-ffhq-config-f", "stylegan2-ffhq-config-f.pt")}
 
 
 
37
 
38
  def get_models():
39
  os.makedirs(model_dir, exist_ok=True)
 
33
 
34
  model_repos = {"e4e": ("akhaliq/JoJoGAN_e4e_ffhq_encode", "e4e_ffhq_encode.pt"),
35
  "dlib": ("akhaliq/jojogan_dlib", "shape_predictor_68_face_landmarks.dat"),
36
+ "base": ("akhaliq/jojogan-stylegan2-ffhq-config-f", "stylegan2-ffhq-config-f.pt"),
37
+ "anime": ("rinong/stylegan-nada-models", "anime.pt"),
38
+ "joker": ("rinong/stylegan-nada-models", "joker.pt")
39
+ }
40
 
41
  def get_models():
42
  os.makedirs(model_dir, exist_ok=True)