Spaces:
Runtime error
Runtime error
Added missing model, re-arranged options
Browse files
app.py
CHANGED
|
@@ -30,29 +30,30 @@ model_repos = {"e4e": ("akhaliq/JoJoGAN_e4e_ffhq_encode", "e4e_ffhq_encode.pt"),
|
|
| 30 |
"dlib": ("akhaliq/jojogan_dlib", "shape_predictor_68_face_landmarks.dat"),
|
| 31 |
"sc_fs3": ("rinong/stylegan-nada-models", "fs3.npy"),
|
| 32 |
"base": ("akhaliq/jojogan-stylegan2-ffhq-config-f", "stylegan2-ffhq-config-f.pt"),
|
| 33 |
-
"
|
| 34 |
"joker": ("rinong/stylegan-nada-models", "joker.pt"),
|
| 35 |
-
"
|
| 36 |
-
"ssj": ("rinong/stylegan-nada-models", "ssj.pt"),
|
| 37 |
"white_walker": ("rinong/stylegan-nada-models", "white_walker.pt"),
|
| 38 |
"zuckerberg": ("rinong/stylegan-nada-models", "zuckerberg.pt"),
|
|
|
|
|
|
|
| 39 |
"cubism": ("rinong/stylegan-nada-models", "cubism.pt"),
|
| 40 |
"disney_princess": ("rinong/stylegan-nada-models", "disney_princess.pt"),
|
| 41 |
"edvard_munch": ("rinong/stylegan-nada-models", "edvard_munch.pt"),
|
| 42 |
"van_gogh": ("rinong/stylegan-nada-models", "van_gogh.pt"),
|
| 43 |
"oil": ("rinong/stylegan-nada-models", "oil.pt"),
|
| 44 |
"rick_morty": ("rinong/stylegan-nada-models", "rick_morty.pt"),
|
| 45 |
-
"
|
| 46 |
-
"crochet": ("rinong/stylegan-nada-models", "crochet.pt"),
|
| 47 |
-
"modigliani": ("rinong/stylegan-nada-models", "modigliani.pt"),
|
| 48 |
"shrek": ("rinong/stylegan-nada-models", "shrek.pt"),
|
| 49 |
-
"sketch": ("rinong/stylegan-nada-models", "sketch.pt"),
|
| 50 |
"thanos": ("rinong/stylegan-nada-models", "thanos.pt"),
|
| 51 |
"ukyioe": ("rinong/stylegan-nada-models", "ukyioe.pt"),
|
| 52 |
"witcher": ("rinong/stylegan-nada-models", "witcher.pt"),
|
|
|
|
|
|
|
| 53 |
"marble": ("rinong/stylegan-nada-models", "marble.pt"),
|
|
|
|
|
|
|
| 54 |
"ghibli": ("rinong/stylegan-nada-models", "ghibli.pt"),
|
| 55 |
-
"grafitti_on_wall": ("rinong/stylegan-nada-models", "grafitti_on_wall.pt"),
|
| 56 |
}
|
| 57 |
|
| 58 |
def get_models():
|
|
|
|
| 30 |
"dlib": ("akhaliq/jojogan_dlib", "shape_predictor_68_face_landmarks.dat"),
|
| 31 |
"sc_fs3": ("rinong/stylegan-nada-models", "fs3.npy"),
|
| 32 |
"base": ("akhaliq/jojogan-stylegan2-ffhq-config-f", "stylegan2-ffhq-config-f.pt"),
|
| 33 |
+
"sketch": ("rinong/stylegan-nada-models", "sketch.pt"),
|
| 34 |
"joker": ("rinong/stylegan-nada-models", "joker.pt"),
|
| 35 |
+
"botero": ("rinong/stylegan-nada-models", "botero.pt"),
|
|
|
|
| 36 |
"white_walker": ("rinong/stylegan-nada-models", "white_walker.pt"),
|
| 37 |
"zuckerberg": ("rinong/stylegan-nada-models", "zuckerberg.pt"),
|
| 38 |
+
"simpson": ("rinong/stylegan-nada-models", "simpson.pt"),
|
| 39 |
+
"ssj": ("rinong/stylegan-nada-models", "ssj.pt"),
|
| 40 |
"cubism": ("rinong/stylegan-nada-models", "cubism.pt"),
|
| 41 |
"disney_princess": ("rinong/stylegan-nada-models", "disney_princess.pt"),
|
| 42 |
"edvard_munch": ("rinong/stylegan-nada-models", "edvard_munch.pt"),
|
| 43 |
"van_gogh": ("rinong/stylegan-nada-models", "van_gogh.pt"),
|
| 44 |
"oil": ("rinong/stylegan-nada-models", "oil.pt"),
|
| 45 |
"rick_morty": ("rinong/stylegan-nada-models", "rick_morty.pt"),
|
| 46 |
+
"anime": ("rinong/stylegan-nada-models", "anime.pt"),
|
|
|
|
|
|
|
| 47 |
"shrek": ("rinong/stylegan-nada-models", "shrek.pt"),
|
|
|
|
| 48 |
"thanos": ("rinong/stylegan-nada-models", "thanos.pt"),
|
| 49 |
"ukyioe": ("rinong/stylegan-nada-models", "ukyioe.pt"),
|
| 50 |
"witcher": ("rinong/stylegan-nada-models", "witcher.pt"),
|
| 51 |
+
"grafitti_on_wall": ("rinong/stylegan-nada-models", "grafitti_on_wall.pt"),
|
| 52 |
+
"modernism": ("rinong/stylegan-nada-models", "modernism.pt"),
|
| 53 |
"marble": ("rinong/stylegan-nada-models", "marble.pt"),
|
| 54 |
+
"crochet": ("rinong/stylegan-nada-models", "crochet.pt"),
|
| 55 |
+
"modigliani": ("rinong/stylegan-nada-models", "modigliani.pt"),
|
| 56 |
"ghibli": ("rinong/stylegan-nada-models", "ghibli.pt"),
|
|
|
|
| 57 |
}
|
| 58 |
|
| 59 |
def get_models():
|