Update app.py

app.py

@@ -159,12 +159,13 @@ with gr.Blocks(css='style.css') as demo:
                )
         with gr.Row():
             with gr.Row():
-                output1=gr.Image(label="ImMagician", show_label=False, min_width=640, object_fit="contain", height="auto")
-                output2=gr.Image(label="ImMagician", show_label=False, min_width=640, object_fit="contain", height="auto")
+                output1=gr.Image(label="ImMagician", show_label=False, min_width=640, object_fit="contain", height="auto", download=True)
+                output2=gr.Image(label="ImMagician", show_label=False, min_width=640, object_fit="contain", height="auto", download=True)
         with gr.Row():
             with gr.Row():
-                output3=gr.Image(label="ImMagician", show_label=False, min_width=640, object_fit="contain", height="auto")
-                output4=gr.Image(label="ImMagician", show_label=False, min_width=640, object_fit="contain", height="auto")
+                output3=gr.Image(label="ImMagician", show_label=False, min_width=640, object_fit="contain", height="auto", download=True)
+                output4=gr.Image(label="ImMagician", show_label=False, min_width=640, object_fit="contain", height="auto", download=True)
+                
         see_prompts.click(get_prompts, inputs=[input_text], outputs=[prompt], queue=False)
         run.click(send_it1, inputs=[prompt, noise_level], outputs=[output1])
         run.click(send_it2, inputs=[prompt, noise_level], outputs=[output2])