Update app.py

app.py

@@ -56,7 +56,6 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, guidan
     ):
         yield img, seed
 
-
 examples = [
     "a man walking in the forest",
     "a viking ship sailing down a river",
@@ -84,9 +83,6 @@ css = """
 """
 
 with gr.Blocks(css=css) as natalie_diffusion:
-    # Predefine result image so it's accessible to both sides
-    result = gr.Image(label="", show_label=False, elem_id="generated-image")
-
     with gr.Row():
         with gr.Column(scale=1, elem_id="left-column"):
             gr.Markdown("""
@@ -119,17 +115,20 @@ Generate images in the surreal style of artist [Natalie Kav](https://www.behance
                     guidance_scale = gr.Slider(label="Guidance Scale", minimum=1, maximum=15, step=0.1, value=3.5)
                     num_inference_steps = gr.Slider(label="Number of inference steps", minimum=1, maximum=50, step=1, value=28)
 
+            # Define hidden result image just for example caching
+            result_example = gr.Image(visible=False)
+
             gr.Examples(
                 examples=examples,
                 fn=infer,
                 inputs=[prompt],
-                outputs=[result, seed],
+                outputs=[result_example, seed],
                 cache_examples=True,
                 cache_mode="lazy"
             )
 
         with gr.Column(scale=1, elem_id="right-column"):
-            result.render()
+            result = gr.Image(label="", show_label=False, elem_id="generated-image")
 
     gr.on(
         triggers=[run_button.click, prompt.submit],