Update app.py

app.py

@@ -61,6 +61,7 @@ def send_it1(inputs, noise_level, proc1=proc1):
         time.sleep(2)
     queue.put(prompt_with_noise)
     output1 = proc1(prompt_with_noise)
+    image.save(f"{prompt_list}.png")
     return output1
 
 
@@ -111,7 +112,17 @@ with gr.Blocks(css='style.css') as demo:
             output2 = gr.Image(label="Dreamlike Diffusion 1.0", show_label=False)
 
     run.click(send_it1, inputs=[prompt, noise_level], outputs=[output1])
-    run.click(send_it2, inputs=[prompt, noise_level+0.2], outputs=[output2])
+    with gr.Row():
+                gr.HTML(
+    """
+        <div class="footer">
+        <p><a href=f"/{prompt_list}.png" download>
+  <img src=f"/{prompt_list}.png">
+</a>
+</p>
+</div>
+    """
+)
 
 demo.launch(enable_queue=True, inline=True)
 block.queue(concurrency_count=100)