Changed visibility settings. Removed some models for quicker debugging

README.md

@@ -4,7 +4,7 @@ emoji: 🌖
 colorFrom: blue
 colorTo: gray
 sdk: gradio
-sdk_version: 3.0b2
+sdk_version: 3.0b6
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -36,20 +36,22 @@ model_repos = {"e4e": ("akhaliq/JoJoGAN_e4e_ffhq_encode", "e4e_ffhq_encode.pt"),
                "base": ("akhaliq/jojogan-stylegan2-ffhq-config-f", "stylegan2-ffhq-config-f.pt"),
                "anime": ("rinong/stylegan-nada-models", "anime.pt"),
                "joker": ("rinong/stylegan-nada-models", "joker.pt"),
-               "simpson": ("rinong/stylegan-nada-models", "simpson.pt"),
-               "ssj": ("rinong/stylegan-nada-models", "ssj.pt"),
-               "white_walker": ("rinong/stylegan-nada-models", "white_walker.pt"),
-               "zuckerberg": ("rinong/stylegan-nada-models", "zuckerberg.pt"),
-               "cubism": ("rinong/stylegan-nada-models", "cubism.pt"),
-               "disney_princess": ("rinong/stylegan-nada-models", "disney_princess.pt"),
-               "edvard_munch": ("rinong/stylegan-nada-models", "edvard_munch.pt"),
-               "van_gogh": ("rinong/stylegan-nada-models", "van_gogh.pt"),
-               "oil": ("rinong/stylegan-nada-models", "oil.pt"),
-               "rick_morty": ("rinong/stylegan-nada-models", "rick_morty.pt"),
-               "botero": ("rinong/stylegan-nada-models", "botero.pt"),
-               "crochet": ("rinong/stylegan-nada-models", "crochet.pt"),
-               "modigliani": ("rinong/stylegan-nada-models", "modigliani.pt"),
-               "shrek": ("rinong/stylegan-nada-models", "shrek.pt"),
+            #    "simpson": ("rinong/stylegan-nada-models", "simpson.pt"),
+            #    "ssj": ("rinong/stylegan-nada-models", "ssj.pt"),
+            #    "white_walker": ("rinong/stylegan-nada-models", "white_walker.pt"),
+            #    "zuckerberg": ("rinong/stylegan-nada-models", "zuckerberg.pt"),
+            #    "cubism": ("rinong/stylegan-nada-models", "cubism.pt"),
+            #    "disney_princess": ("rinong/stylegan-nada-models", "disney_princess.pt"),
+            #    "edvard_munch": ("rinong/stylegan-nada-models", "edvard_munch.pt"),
+            #    "van_gogh": ("rinong/stylegan-nada-models", "van_gogh.pt"),
+            #    "oil": ("rinong/stylegan-nada-models", "oil.pt"),
+            #    "rick_morty": ("rinong/stylegan-nada-models", "rick_morty.pt"),
+            #    "botero": ("rinong/stylegan-nada-models", "botero.pt"),
+            #    "crochet": ("rinong/stylegan-nada-models", "crochet.pt"),
+            #    "modigliani": ("rinong/stylegan-nada-models", "modigliani.pt"),
+            #    "shrek": ("rinong/stylegan-nada-models", "shrek.pt"),
+            #    "sketch": ("rinong/stylegan-nada-models", "sketch.pt"),
+            #    "thanos": ("rinong/stylegan-nada-models", "thanos.pt"),
                }
 
 def get_models():
@@ -228,15 +230,15 @@ with blocks:
         with gr.Row():
             video_choice = gr.inputs.Checkbox(default=False, label="Generate Video?", optional=False)
             
-            edit_choice       = gr.inputs.Checkbox(default=False, label="With Editing?", optional=False, visible=False)
-            vid_format_choice = gr.inputs.Radio(choices=["gif", "mp4"], type="value", default='mp4', label="Video Format", visible=False)
+            edit_choice       = gr.inputs.Checkbox(default=False, label="With Editing?", optional=False)
+            vid_format_choice = gr.inputs.Radio(choices=["gif", "mp4"], type="value", default='mp4', label="Video Format")
         with gr.Row():
             img_button = gr.Button("Edit Image")
             vid_button = gr.Button("Edit Image")
 
     with gr.Column():
         img_output = gr.outputs.Image(type="file")
-        vid_output = gr.outputs.Video(visible=False)
+        vid_output = gr.outputs.Video()
 
     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])
     video_choice.change(fn=change_component_visibility, inputs=[video_choice, [img_output, img_button], True], outputs=[img_output, img_button])