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)
-    result = gr.Gallery(label="Result", columns=1, preview=True, show_label=False)
+        run_button = gr.Button("Run", scale=0, variant="primary")
+    result = gr.Image(label="Result", columns=1, preview=True, show_label=False)
         
     with gr.Row(visible=True):
         grid_size_selection = gr.Dropdown(
@@ -209,7 +209,8 @@ with gr.Blocks(css=css, theme="prithivMLmods/Minecraft-Theme") as demo:
         examples=examples, 
         inputs=[prompt], 
         outputs=[result, seed], 
-        fn=infer
+        fn=infer,
+        cache_examples=True
     )
         
     gr.on(