Update app.py

app.py

@@ -118,18 +118,18 @@ with gr.Blocks(css=css) as demo:
             negative_prompt = gr.Textbox(label="Negative Prompt", value="text, blurry, fuzziness", placeholder="What you don't want to see in the image", show_label=False, lines=3)
         with gr.Column():
             text_button = gr.Button("Generate", variant='primary', elem_id="generate")
+            
     with gr.Row():
-        with gr.Column(scale=6):
-            model = gr.Dropdown(interactive=True,value="v1-5-pruned-emaonly.safetensors [d7049739]", show_label=True, label="Model", choices=prodia_client.list_models())
-
-    with gr.Row():
-        with gr.Column(scale=3):
+        
             
                 
         with gr.Column(scale=2):
-            image_output = gr.Gallery(label="Result Image(s)", show_download_button=False, show_share_button=True)
-
+            image_output = gr.Image()
+            
         with gr.Accordion("Advanced options", open=False):
+                    with gr.Column(scale=6):
+                        model = gr.Dropdown(interactive=True,value="v1-5-pruned-emaonly.safetensors [d7049739]", show_label=True, label="Model", choices=prodia_client.list_models())
+
                     with gr.Column(scale=1):
                         sampler = gr.Dropdown(value="DPM++ SDE", show_label=True, label="Sampler", choices=[
                             "Euler",