rinong commited on
Commit
4ec364e
β€’
1 Parent(s): 8e301b6

Changed visibility settings. Removed some models for quicker debugging

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +19 -17
README.md CHANGED
@@ -4,7 +4,7 @@ emoji: πŸŒ–
4
  colorFrom: blue
5
  colorTo: gray
6
  sdk: gradio
7
- sdk_version: 3.0b2
8
  app_file: app.py
9
  pinned: false
10
  license: mit
 
4
  colorFrom: blue
5
  colorTo: gray
6
  sdk: gradio
7
+ sdk_version: 3.0b6
8
  app_file: app.py
9
  pinned: false
10
  license: mit
app.py CHANGED
@@ -36,20 +36,22 @@ model_repos = {"e4e": ("akhaliq/JoJoGAN_e4e_ffhq_encode", "e4e_ffhq_encode.pt"),
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
- "simpson": ("rinong/stylegan-nada-models", "simpson.pt"),
40
- "ssj": ("rinong/stylegan-nada-models", "ssj.pt"),
41
- "white_walker": ("rinong/stylegan-nada-models", "white_walker.pt"),
42
- "zuckerberg": ("rinong/stylegan-nada-models", "zuckerberg.pt"),
43
- "cubism": ("rinong/stylegan-nada-models", "cubism.pt"),
44
- "disney_princess": ("rinong/stylegan-nada-models", "disney_princess.pt"),
45
- "edvard_munch": ("rinong/stylegan-nada-models", "edvard_munch.pt"),
46
- "van_gogh": ("rinong/stylegan-nada-models", "van_gogh.pt"),
47
- "oil": ("rinong/stylegan-nada-models", "oil.pt"),
48
- "rick_morty": ("rinong/stylegan-nada-models", "rick_morty.pt"),
49
- "botero": ("rinong/stylegan-nada-models", "botero.pt"),
50
- "crochet": ("rinong/stylegan-nada-models", "crochet.pt"),
51
- "modigliani": ("rinong/stylegan-nada-models", "modigliani.pt"),
52
- "shrek": ("rinong/stylegan-nada-models", "shrek.pt"),
 
 
53
  }
54
 
55
  def get_models():
@@ -228,15 +230,15 @@ with blocks:
228
  with gr.Row():
229
  video_choice = gr.inputs.Checkbox(default=False, label="Generate Video?", optional=False)
230
 
231
- edit_choice = gr.inputs.Checkbox(default=False, label="With Editing?", optional=False, visible=False)
232
- vid_format_choice = gr.inputs.Radio(choices=["gif", "mp4"], type="value", default='mp4', label="Video Format", visible=False)
233
  with gr.Row():
234
  img_button = gr.Button("Edit Image")
235
  vid_button = gr.Button("Edit Image")
236
 
237
  with gr.Column():
238
  img_output = gr.outputs.Image(type="file")
239
- vid_output = gr.outputs.Video(visible=False)
240
 
241
  video_choice.change(fn=change_component_visibility, inputs=[video_choice, [edit_choice, vid_format_choice, vid_output, vid_button]], outputs=[edit_choice, vid_format_choice, vid_output, vid_button])
242
  video_choice.change(fn=change_component_visibility, inputs=[video_choice, [img_output, img_button], True], outputs=[img_output, img_button])
 
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
+ # "simpson": ("rinong/stylegan-nada-models", "simpson.pt"),
40
+ # "ssj": ("rinong/stylegan-nada-models", "ssj.pt"),
41
+ # "white_walker": ("rinong/stylegan-nada-models", "white_walker.pt"),
42
+ # "zuckerberg": ("rinong/stylegan-nada-models", "zuckerberg.pt"),
43
+ # "cubism": ("rinong/stylegan-nada-models", "cubism.pt"),
44
+ # "disney_princess": ("rinong/stylegan-nada-models", "disney_princess.pt"),
45
+ # "edvard_munch": ("rinong/stylegan-nada-models", "edvard_munch.pt"),
46
+ # "van_gogh": ("rinong/stylegan-nada-models", "van_gogh.pt"),
47
+ # "oil": ("rinong/stylegan-nada-models", "oil.pt"),
48
+ # "rick_morty": ("rinong/stylegan-nada-models", "rick_morty.pt"),
49
+ # "botero": ("rinong/stylegan-nada-models", "botero.pt"),
50
+ # "crochet": ("rinong/stylegan-nada-models", "crochet.pt"),
51
+ # "modigliani": ("rinong/stylegan-nada-models", "modigliani.pt"),
52
+ # "shrek": ("rinong/stylegan-nada-models", "shrek.pt"),
53
+ # "sketch": ("rinong/stylegan-nada-models", "sketch.pt"),
54
+ # "thanos": ("rinong/stylegan-nada-models", "thanos.pt"),
55
  }
56
 
57
  def get_models():
 
230
  with gr.Row():
231
  video_choice = gr.inputs.Checkbox(default=False, label="Generate Video?", optional=False)
232
 
233
+ edit_choice = gr.inputs.Checkbox(default=False, label="With Editing?", optional=False)
234
+ vid_format_choice = gr.inputs.Radio(choices=["gif", "mp4"], type="value", default='mp4', label="Video Format")
235
  with gr.Row():
236
  img_button = gr.Button("Edit Image")
237
  vid_button = gr.Button("Edit Image")
238
 
239
  with gr.Column():
240
  img_output = gr.outputs.Image(type="file")
241
+ vid_output = gr.outputs.Video()
242
 
243
  video_choice.change(fn=change_component_visibility, inputs=[video_choice, [edit_choice, vid_format_choice, vid_output, vid_button]], outputs=[edit_choice, vid_format_choice, vid_output, vid_button])
244
  video_choice.change(fn=change_component_visibility, inputs=[video_choice, [img_output, img_button], True], outputs=[img_output, img_button])