rinong commited on
Commit
d48a8f6
1 Parent(s): 4f361bc

Added missing model, re-arranged options

Browse files
Files changed (1) hide show
  1. app.py +9 -8
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
- "anime": ("rinong/stylegan-nada-models", "anime.pt"),
34
  "joker": ("rinong/stylegan-nada-models", "joker.pt"),
35
- "simpson": ("rinong/stylegan-nada-models", "simpson.pt"),
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
- "botero": ("rinong/stylegan-nada-models", "botero.pt"),
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():