Update app.py

app.py

@@ -50,7 +50,7 @@ with gr.Blocks(css=css) as app:
             image_output3 = gr.Image(label="Generated Image 3")
             image_output4 = gr.Image(label="Generated Image 4")
 
-    generate_button.click(generate_image_from_text, inputs=[text_prompt, model_selector], outputs=[image_output1, image_output2, image_output3, image_output4])
+    generate_button.click(generate_image_from_text_with_persistent_storage, inputs=[text_prompt, model_selector], outputs=[image_output1, image_output2, image_output3, image_output4])
 
 # Launch the app
 app.launch()