rinong commited on
Commit
cb4b687
1 Parent(s): 951f6cb

Added a bunch of models

Browse files
Files changed (1) hide show
  1. app.py +21 -28
app.py CHANGED
@@ -32,22 +32,27 @@ model_repos = {"e4e": ("akhaliq/JoJoGAN_e4e_ffhq_encode", "e4e_ffhq_encode.pt"),
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
  }
52
 
53
  def get_models():
@@ -290,18 +295,6 @@ class ImageEditor(object):
290
 
291
  editor = ImageEditor()
292
 
293
- # def change_component_visibility(component_types, invert_choices):
294
-
295
- # def visibility_impl(visible):
296
- # return [component_types[idx].update(visible=visible ^ invert_choices[idx]) for idx in range(len(component_types))]
297
-
298
- # return visibility_impl
299
-
300
- # def group_visibility(visible):
301
- # print("visible: ", visible)
302
-
303
- # return gr.Group.update(visibile=visible)
304
-
305
  blocks = gr.Blocks()
306
 
307
  with blocks:
 
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():
 
295
 
296
  editor = ImageEditor()
297
 
 
 
 
 
 
 
 
 
 
 
 
 
298
  blocks = gr.Blocks()
299
 
300
  with blocks: