Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import numpy as np
 import random
+import time
 from optimum.intel import OVStableDiffusionXLPipeline
 import torch
 from diffusers import EulerDiscreteScheduler
@@ -69,6 +70,9 @@ def infer(
         generator=generator,
     ).images[0]
 
+    filename = f"./tmp/{time.time()}.jpg"
+    image.save(filename)
+
     return image, seed
 
 
@@ -96,6 +100,7 @@ with gr.Blocks() as img:
 
             run_button = gr.Button("Run", scale=0, variant="primary")
 
+        file_output = gr.File(label="Download your image")
         result = gr.Image(label="Result", show_label=False)
 
         with gr.Accordion("Advanced Settings", open=False):
@@ -169,7 +174,7 @@ with gr.Blocks() as img:
             guidance_scale,
             num_inference_steps,
         ],
-        outputs=[result, seed],
+        outputs=[result, seed, file_output],
     )
 
 if __name__ == "__main__":