UI update

app.py

@@ -48,6 +48,7 @@ model_repos = {"e4e": ("akhaliq/JoJoGAN_e4e_ffhq_encode", "e4e_ffhq_encode.pt"),
                "shrek": ("rinong/stylegan-nada-models", "shrek.pt"),
                "thanos": ("rinong/stylegan-nada-models", "thanos.pt"),
                "ukiyoe": ("rinong/stylegan-nada-models", "ukiyoe.pt"),
+               "groot": ("rinong/stylegan-nada-models", "groot.pt"),
                "witcher": ("rinong/stylegan-nada-models", "witcher.pt"),
                "grafitti_on_wall": ("rinong/stylegan-nada-models", "grafitti_on_wall.pt"),
                "modernism": ("rinong/stylegan-nada-models", "modernism.pt"),
@@ -321,7 +322,22 @@ with blocks:
     gr.Markdown("<h4 style='font-size: 110%;margin-top:.5em'>A note on social impact</h4><div>This model relies on StyleGAN and CLIP, both of which are prone to biases inherited from their training data and their architecture. These may include (but are not limited to) poor representation of minorities or the perpetution of societal biases, such as gender norms. In particular, StyleGAN editing may induce undesired changes in skin tones. Moreover, generative models can, and have been used to create deep fake imagery which may assist in the spread of propaganda. However, <a href='https://github.com/NVlabs/stylegan3-detector' target='_blank'>tools are available</a> for identifying StyleGAN generated imagery, and any 'realistic' results produced by this model should be easily identifiable through such tools.</div>")
     
     with gr.Row():
-        input_img = gr.inputs.Image(type="filepath", label="Input image")
+        with gr.Column():
+            input_img = gr.inputs.Image(type="filepath", label="Input image")
+            
+            with gr.Tabs():
+                with gr.TabItem("Edit Images"):
+                    img_button = gr.Button("Edit Image")
+                    img_output = gr.Gallery(label="Output Images")
+                    
+                with gr.TabItem("Create Video"):
+                    with gr.Row():
+                        vid_button  = gr.Button("Generate Video")
+                        loop_styles = gr.inputs.Checkbox(default=True, label="Loop video back to the initial style?")
+                    with gr.Row():
+                            gr.Markdown("Warning: Videos generation requires the synthesis of hundreds of frames and is expected to take several minutes.")
+                            gr.Markdown("To reduce queue times, we significantly reduced the number of video frames. Using more than 3 styles will further reduce the frames per style, leading to quicker transitions. For better control, we recommend cloning the gradio app, adjusting <b>num_alphas</b> in <b>generate_videos.py</b>, and running the code locally.")
+                    vid_output = gr.outputs.Video(label="Output Video")
 
         with gr.Column():    
             style_choice = gr.inputs.CheckboxGroup(choices=editor.get_style_list(), type="value", label="Choose your styles!")
@@ -357,28 +373,6 @@ with blocks:
 
                     sc_edit_choices = [src_text_styleclip, tar_text_styleclip, alpha_styleclip, beta_styleclip]
 
-    with gr.Tabs():
-        with gr.TabItem("Edit Images"):
-            with gr.Row():
-                with gr.Column():
-                    with gr.Row():
-                        img_button = gr.Button("Edit Image")
-                with gr.Column():
-                    img_output = gr.Gallery(label="Output Images")
-            
-        with gr.TabItem("Create Video"):
-            with gr.Row():
-                with gr.Column():
-                    with gr.Row():
-                        vid_button  = gr.Button("Generate Video")
-                        loop_styles = gr.inputs.Checkbox(default=True, label="Loop video back to the initial style?")
-                    with gr.Row():
-                        with gr.Column():
-                            gr.Markdown("Warning: Videos generation requires the synthesis of hundreds of frames and is expected to take several minutes.")
-                            gr.Markdown("To reduce queue times, we significantly reduced the number of video frames. Using more than 3 styles will further reduce the frames per style, leading to quicker transitions. For better control, we recommend cloning the gradio app, adjusting <b>num_alphas</b> in <b>generate_videos.py</b>, and running the code locally.")
-                with gr.Column():
-                    vid_output = gr.outputs.Video(label="Output Video")
-
     edit_inputs = [editing_type_choice] + ig_edit_choices + sc_edit_choices
     img_button.click(fn=editor.edit_image, inputs=edit_inputs + [input_img, style_choice], outputs=img_output)
     vid_button.click(fn=editor.edit_video, inputs=edit_inputs + [input_img, style_choice, loop_styles], outputs=vid_output)