Stable Diffusion 3 Medium

app.py

@@ -10,6 +10,7 @@ import time
 import random
 import spaces
 import re
+import uuid
 
 from gradio_imageslider import ImageSlider
 from PIL import Image
@@ -377,6 +378,9 @@ def restore(
     " pixels large and " + str(result_height) + \
     " pixels high, so a resolution of " + f'{result_width * result_height:,}' + " pixels."
     print(information)
+    unique_name = str(uuid.uuid4()) + ".png"
+    results[0].save(unique_name)
+    print(unique_name)
 
     # Only one image can be shown in the slider
     return [noisy_image] + [results[0]], gr.update(label="Downloadable results in *." + output_format + " format", format = output_format, value = results), gr.update(value = information, visible = True)
@@ -437,7 +441,7 @@ title_html = """
     SUPIR is for beauty and illustration only.
     Most of the processes only last few minutes.
     This demo can handle huge images but the process will be aborted if it lasts more than 10 min.
-    If you want to upscale AI-generated images, be noticed that <i>Stable Diffusion Medium</i> spaces can directly generate 1536x1536 images.
+    If you want to upscale AI-generated images, be noticed that <i>Stable Diffusion 3 Medium</i> spaces can directly generate 1536x1536 images.
     Please leave a <a href="https://huggingface.co/spaces/Fabrice-TIERCELIN/SUPIR/discussions/new">message in discussion</a> if you encounter issues.
     
     <p><center><a href="https://arxiv.org/abs/2401.13627">Paper</a> &emsp; <a href="http://supir.xpixel.group/">Project Page</a> &emsp; <a href="https://huggingface.co/blog/MonsterMMORPG/supir-sota-image-upscale-better-than-magnific-ai">Local Install Guide</a></center></p>