Update app.py

app.py

@@ -97,49 +97,87 @@ options = ['SD1.4', 'SD1.5', 'SD2.1']
 with block:
     gr.Markdown("SD vs. FreeU.")
     with gr.Row():
-        with gr.Column():
-            sd_options = gr.Dropdown(options, value='SD1.4', label="SD options")
-            model_id = "CompVis/stable-diffusion-v1-4"
-            with gr.Accordion('FreeU Parameters', open=False):
-                b1 = gr.Slider(label='b1: backbone factor of the first stage block of decoder',
-                                        minimum=1,
-                                        maximum=1.6,
-                                        step=0.1,
-                                        value=1)
-                b2 = gr.Slider(label='b2: backbone factor of the second stage block of decoder',
-                                        minimum=1,
-                                        maximum=1.6,
-                                        step=0.1,
-                                        value=1)
-                s1 = gr.Slider(label='s1: skip factor of the first stage block of decoder',
-                                        minimum=0,
-                                        maximum=1,
-                                        step=0.1,
-                                        value=1)
-                s2 = gr.Slider(label='s2: skip factor of the second stage block of decoder',
-                                        minimum=0,
-                                        maximum=1,
-                                        step=0.1,
-                                        value=1)
-    with gr.Group():
-        with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
-            with gr.Column():
-                text = gr.Textbox(
-                    label="Enter your prompt",
-                    show_label=False,
-                    max_lines=1,
-                    placeholder="Enter your prompt",
-                    container=False,
-                )
-            btn = gr.Button("Generate image", scale=0)
+        with gr.Group():
+            with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
+                with gr.Column():
+                    text = gr.Textbox(
+                        label="Enter your prompt",
+                        show_label=False,
+                        max_lines=1,
+                        placeholder="Enter your prompt",
+                        container=False,
+                    )
+                btn = gr.Button("Generate image", scale=0)
+
+            with gr.Row():
+                with gr.Column(min_width=256) as c1:
+                    image_1 = gr.Image(interactive=False)
+                    image_1_label = gr.Markdown("SD")
+                with gr.Row():
+            
+            
+        
+        with gr.Group():
+            with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
+                with gr.Column():
+                    text = gr.Textbox(
+                        label="Enter your prompt",
+                        show_label=False,
+                        max_lines=1,
+                        placeholder="Enter your prompt",
+                        container=False,
+                    )
+                btn = gr.Button("Generate image", scale=0)
+
+            with gr.Row():
+                with gr.Column(min_width=256) as c2:
+                    image_2 = gr.Image(interactive=False)
+                    image_2_label = gr.Markdown("FreeU")
+        
+        
+        # with gr.Column():
+        #     sd_options = gr.Dropdown(options, value='SD1.4', label="SD options")
+        #     model_id = "CompVis/stable-diffusion-v1-4"
+        #     with gr.Accordion('FreeU Parameters', open=False):
+        #         b1 = gr.Slider(label='b1: backbone factor of the first stage block of decoder',
+        #                                 minimum=1,
+        #                                 maximum=1.6,
+        #                                 step=0.1,
+        #                                 value=1)
+        #         b2 = gr.Slider(label='b2: backbone factor of the second stage block of decoder',
+        #                                 minimum=1,
+        #                                 maximum=1.6,
+        #                                 step=0.1,
+        #                                 value=1)
+        #         s1 = gr.Slider(label='s1: skip factor of the first stage block of decoder',
+        #                                 minimum=0,
+        #                                 maximum=1,
+        #                                 step=0.1,
+        #                                 value=1)
+        #         s2 = gr.Slider(label='s2: skip factor of the second stage block of decoder',
+        #                                 minimum=0,
+        #                                 maximum=1,
+        #                                 step=0.1,
+        #                                 value=1)
+    # with gr.Group():
+    #     with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
+    #         with gr.Column():
+    #             text = gr.Textbox(
+    #                 label="Enter your prompt",
+    #                 show_label=False,
+    #                 max_lines=1,
+    #                 placeholder="Enter your prompt",
+    #                 container=False,
+    #             )
+    #         btn = gr.Button("Generate image", scale=0)
         
-        with gr.Row():
-            with gr.Column(min_width=256) as c1:
-                image_1 = gr.Image(interactive=False)
-                image_1_label = gr.Markdown("SD")
-            with gr.Column(min_width=256) as c2:
-                image_2 = gr.Image(interactive=False)
-                image_2_label = gr.Markdown("FreeU")
+    #     with gr.Row():
+    #         with gr.Column(min_width=256) as c1:
+    #             image_1 = gr.Image(interactive=False)
+    #             image_1_label = gr.Markdown("SD")
+    #         with gr.Column(min_width=256) as c2:
+    #             image_2 = gr.Image(interactive=False)
+    #             image_2_label = gr.Markdown("FreeU")
             
 
     ex = gr.Examples(examples=examples, fn=infer, inputs=[text], outputs=[image_1, image_2], cache_examples=False)