Update app.py

app.py

@@ -1,4 +1,5 @@
 import os
+import datetime
 import einops
 import gradio as gr
 from gradio_imageslider import ImageSlider
@@ -74,6 +75,9 @@ def inference(input_image, prompt, a_prompt, n_prompt, denoise_steps, upscale, a
     resize_preproc = transforms.Compose([
         transforms.Resize(process_size, interpolation=transforms.InterpolationMode.BILINEAR),
     ])
+    
+    # Get the current timestamp
+    timestamp = datetime.datetime.now().strftime("%Y%m%d%H%M%S")
 
     with torch.no_grad():
         seed_everything(seed)
@@ -119,12 +123,12 @@ def inference(input_image, prompt, a_prompt, n_prompt, denoise_steps, upscale, a
             image = Image.new(mode="RGB", size=(512, 512))
     
     # Convert and save the image as JPEG
-    image.save('result.jpg', 'JPEG')
+    image.save(f'result_{timestamp}.jpg', 'JPEG')
 
     # Convert and save the image as JPEG
-    input_image.save('input.jpg', 'JPEG')
+    input_image.save(f'input_{timestamp}.jpg', 'JPEG')
     
-    return ("input.jpg", "result.jpg"), "result.jpg"
+    return (f"input_{timestamp}.jpg", f"result_{timestamp}.jpg"), "result.jpg"
 
 title = "Pixel-Aware Stable Diffusion for Real-ISR"
 description = "Gradio Demo for PASD Real-ISR. To use it, simply upload your image, or click one of the examples to load them."