Update app.py

app.py

@@ -117,17 +117,19 @@ def generate(
 
 with gr.Blocks(css=css, theme=gr.themes.Soft()) as demo:
     gr.Markdown(DESCRIPTIONx)
-    with gr.Group():
-        with gr.Row():
-            prompt = gr.Text(
-                label="Prompt",
-                show_label=False,
-                max_lines=1,
-                placeholder="Enter your prompt",
-                container=False,
-            )
-            run_button = gr.Button("Run", scale=0)
-        result = gr.Gallery(label="Result", columns=2, show_label=False)
+    with gr.Row():
+        with gr.Column(scale=1):
+            with gr.Row():
+                prompt = gr.Text(
+                    show_label=False,
+                    max_lines=1,
+                    placeholder="Enter your prompt",
+                    container=False,
+                )
+                run_button = gr.Button("Run", scale=0, variant="primary")
+
+            result = gr.Image(show_label=False, format="png")
+            
     with gr.Accordion("Advanced options", open=False, visible=True):
         num_images = gr.Slider(
             label="Number of Images",