Update app.py

app.py

@@ -824,6 +824,12 @@ hardcoded_prompt_2="A vibrant and dynamic football game scene in the style of Pe
 hardcoded_prompt_3 = "Create a high-energy scene of a DJ performing on a large stage with vibrant lights, colorful lasers, a lively dancing crowd, and various electronic equipment in the background."
 
 
+def update_images():
+    image_1 = generate_image(hardcoded_prompt_1)
+    image_2 = generate_image(hardcoded_prompt_2)
+    image_3 = generate_image(hardcoded_prompt_3)
+    return image_1, image_2, image_3
+
 with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
     
     with gr.Row():
@@ -861,6 +867,10 @@ with gr.Blocks(theme='Pijush2023/scikit-learn-pijush') as demo:
             image_output_2 = gr.Image(value=generate_image(hardcoded_prompt_2), width=400, height=400)
             image_output_3 = gr.Image(value=generate_image(hardcoded_prompt_3), width=400, height=400)
 
+
+            refresh_button = gr.Button("Refresh Images")
+            refresh_button.click(fn=update_images, inputs=None, outputs=[image_output_1, image_output_2, image_output_3])
+
 demo.queue()
 demo.launch(share=True)