Update app.py

app.py

@@ -144,18 +144,17 @@ def generate(
 #Main gr.Block
 with gr.Blocks(css=css, theme="YTheme/Minecraft") 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=1, show_label=False) 
+    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=1, preview=True, show_label=False)
+    
     with gr.Accordion("Advanced options", open=False, visible=False):
         num_images = gr.Slider(
             label="Number of Images",