Only one result can be shown in the slider

gradio_demo.py

@@ -28,7 +28,7 @@ parser.add_argument("--opt", type=str, default='options/SUPIR_v0.yaml')
 parser.add_argument("--ip", type=str, default='127.0.0.1')
 parser.add_argument("--port", type=int, default='6688')
 parser.add_argument("--no_llava", action='store_true', default=True)#False
-parser.add_argument("--use_image_slider", action='store_true', default=True)#False
+parser.add_argument("--use_image_slider", action='store_true', default=False)#False
 parser.add_argument("--log_history", action='store_true', default=False)
 parser.add_argument("--loading_half_params", action='store_true', default=False)#False
 parser.add_argument("--use_tile_vae", action='store_true', default=True)#False
@@ -78,7 +78,7 @@ def check(input_image):
 def reset_feedback():
     return 3, ''
 
-@spaces.GPU(duration=240)
+@spaces.GPU(duration=300)
 def stage1_process(input_image, gamma_correction):
     print('Start stage1_process')
     if torch.cuda.device_count() == 0:
@@ -100,7 +100,7 @@ def stage1_process(input_image, gamma_correction):
     print('End stage1_process')
     return LQ, gr.update(visible = True)
 
-@spaces.GPU(duration=240)
+@spaces.GPU(duration=300)
 def llave_process(input_image, temperature, top_p, qs=None):
     print('Start llave_process')
     if torch.cuda.device_count() == 0:
@@ -116,7 +116,7 @@ def llave_process(input_image, temperature, top_p, qs=None):
     print('End llave_process')
     return captions[0]
 
-@spaces.GPU(duration=240)
+@spaces.GPU(duration=300)
 def stage2_process(
     noisy_image,
     denoise_image,
@@ -223,7 +223,8 @@ def stage2_process(
     ((str(minutes) + " min, ") if hours != 0 or minutes != 0 else "") + \
     str(secondes) + " sec."
 
-    return [noisy_image] + results, [noisy_image] + results, gr.update(value = information, visible = True), event_id
+    # Only one image can be shown in the slider
+    return [noisy_image] + [results[0]], [noisy_image] + results, gr.update(value = information, visible = True), event_id
 
 def load_and_reset(param_setting):
     print('Start load_and_reset')
@@ -284,7 +285,7 @@ title_html = """
     LlaVa is not integrated in this demo. The content added by SUPIR is imagination, not real-world information.
     The aim of SUPIR is the beauty and the illustration.
     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 an hour.
+    This demo can handle huge images but the process will be aborted if it lasts more than 4min.
     
     <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://github.com/Fanghua-Yu/SUPIR/blob/master/assets/DemoGuide.png">How to play</a> &emsp; <a href="https://huggingface.co/blog/MonsterMMORPG/supir-sota-image-upscale-better-than-magnific-ai">Local Install Guide</a></center></p>
     """