Your computer must not enter into standby mode

app.py

@@ -99,9 +99,10 @@ def reset():
         6
     ]
 
-def check(input_image):
+def check_and_update(input_image):
     if input_image is None:
         raise gr.Error("Please provide an image to restore.")
+    return gr.update(visible = True)
 
 @spaces.GPU(duration=420)
 def stage1_process(
@@ -610,6 +611,7 @@ with gr.Blocks() as interface:
         diffusion_button = gr.Button(value="🚀 Upscale/Restore", variant = "primary", elem_id = "process_button")
         reset_btn = gr.Button(value="🧹 Reinit page", variant="stop", elem_id="reset_button", visible = False)
 
+        warning = gr.HTML(value = "<center><big>Your computer must <u>not</u> enter into standby mode.</big><br/>On Chrome, you can force to keep a tab alive in <code>chrome://discards/</code></center>", visible = False)
         restore_information = gr.HTML(value = "Restart the process to get another result.", visible = False)
         result_slider = ImageSlider(label = 'Comparator', show_label = False, interactive = False, elem_id = "slider1", show_download_button = False)
         result_gallery = gr.Gallery(label = 'Downloadable results', show_label = True, interactive = False, elem_id = "gallery1")
@@ -627,7 +629,7 @@ with gr.Blocks() as interface:
                     1024,
                     1,
                     8,
-                    200,
+                    100,
                     -1,
                     1,
                     7.5,
@@ -792,9 +794,9 @@ with gr.Blocks() as interface:
         rotation
     ], queue = False, show_progress = False)
 
-    denoise_button.click(fn = check, inputs = [
+    denoise_button.click(fn = check_and_update, inputs = [
         input_image
-    ], outputs = [], queue = False, show_progress = False).success(fn = stage1_process, inputs = [
+    ], outputs = [warning], queue = False, show_progress = False).success(fn = stage1_process, inputs = [
         input_image,
         gamma_correction,
         diff_dtype,
@@ -809,9 +811,9 @@ with gr.Blocks() as interface:
         seed
     ], outputs = [
         seed
-    ], queue = False, show_progress = False).then(fn = check, inputs = [
+    ], queue = False, show_progress = False).then(fn = check_and_update, inputs = [
         input_image
-    ], outputs = [], queue = False, show_progress = False).success(fn=stage2_process, inputs = [
+    ], outputs = [warning], queue = False, show_progress = False).success(fn=stage2_process, inputs = [
         input_image,
         rotation,
         denoise_image,