Update app.py

app.py

@@ -97,15 +97,18 @@ options = ['SD1.4', 'SD1.5', 'SD2.1']
 with block:
     gr.Markdown("SD vs. FreeU.")
     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,
-                    )
+        with gr.Row(): 
+            sd_options = gr.Dropdown(['SD1.4', 'SD1.5', 'SD2.1'], value='SD1.4', label="SD options")
+            
+            with gr.Column():
+                text = gr.Textbox(
+                    label="Enter your prompt",
+                    show_label=False,
+                    max_lines=1,
+                    placeholder="Enter your prompt",
+                    container=False,
+                )
+            
                     
     with gr.Row():
         with gr.Group():
@@ -118,7 +121,7 @@ with block:
             #             placeholder="Enter your prompt",
             #             container=False,
             #         )
-                btn = gr.Button("Generate image", scale=0)
+            btn = gr.Button("Generate image", scale=0)
     
             with gr.Row():
                 with gr.Column(min_width=256) as c1:
@@ -126,7 +129,7 @@ with block:
                     image_1_label = gr.Markdown("SD")
             
         with gr.Group():
-            with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
+            # with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
                 # with gr.Column():
                 #     text = gr.Textbox(
                 #         label="Enter your prompt",
@@ -135,7 +138,7 @@ with block:
                 #         placeholder="Enter your prompt",
                 #         container=False,
                 #     )
-                btn = gr.Button("Generate image", scale=0)
+            btn = gr.Button("Generate image", scale=0)
     
             with gr.Row():
                 with gr.Column(min_width=256) as c2: