Update app.py

app.py

@@ -133,8 +133,8 @@ with gr.Blocks(css=css, theme="prithivMLmods/Minecraft-Theme") as demo:
             placeholder="Enter your prompt",
             container=False,
         )
-        run_button = gr.Button("Run", scale=0, variant="primary")
-        result = gr.Image(label="Result", show_label=False)
+        run_button = gr.Button("Run", scale=0)
+    result = gr.Gallery(label="Result", columns=1, preview=True, show_label=False)
         
     with gr.Row(visible=True):
         grid_size_selection = gr.Dropdown(