print('Exception of type ' + str(type(e)))

app.py

@@ -106,7 +106,14 @@ def stage1_process(
     print('<<== stage1_process')
     return LQ, gr.update(visible = True)
 
-def stage2_process(
+def stage2_process(*args, **kwargs):
+    try:
+        return restore_in_Xmin(*args, **kwargs)
+    except Exception as e:
+        print('Exception of type ' + str(type(e)))
+        raise e
+
+def restore_in_Xmin(
     noisy_image,
     rotation,
     denoise_image,
@@ -439,6 +446,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.
     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>