Changes from prod

gradio_demo.py

@@ -79,7 +79,7 @@ def check(input_image):
 def reset_feedback():
     return 3, ''
 
-@spaces.GPU(duration=600)
+@spaces.GPU(duration=540)
 def stage1_process(input_image, gamma_correction):
     print('stage1_process ==>>')
     if torch.cuda.device_count() == 0:
@@ -101,7 +101,7 @@ def stage1_process(input_image, gamma_correction):
     print('<<== stage1_process')
     return LQ, gr.update(visible = True)
 
-@spaces.GPU(duration=600)
+@spaces.GPU(duration=540)
 def llave_process(input_image, temperature, top_p, qs=None):
     print('llave_process ==>>')
     if torch.cuda.device_count() == 0:
@@ -201,7 +201,7 @@ def stage2_process(
 
     return result_slider, result_gallery, restore_information, event_id
 
-@spaces.GPU(duration=600)
+@spaces.GPU(duration=540)
 def restore(
     model,
     edm_steps,
@@ -336,7 +336,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 10 min.
+    This demo can handle huge images but the process will be aborted if it lasts more than 9 min.
     
     <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>
     """
@@ -366,7 +366,6 @@ with gr.Blocks(title="SUPIR") as interface:
     input_image = gr.Image(label="Input", show_label=True, type="numpy", height=600, elem_id="image-input")
     with gr.Group():
         prompt = gr.Textbox(label="Image description for LlaVa", value="", placeholder="A person, walking, in a town, Summer, photorealistic", lines=3, visible=False)
-        upscale = gr.Radio([["x1", 1], ["x2", 2], ["x3", 3], ["x4", 4], ["x5", 5], ["x6", 6], ["x7", 7], ["x8", 8]], label="Upscale factor", info="Resolution x1 to x8", value=2, interactive=True)
         a_prompt = gr.Textbox(label="Image description",
                               info="Help the AI understand what the image represents; describe as much as possible",
                               value='Cinematic, High Contrast, highly detailed, taken using a Canon EOS R '
@@ -375,6 +374,7 @@ with gr.Blocks(title="SUPIR") as interface:
                                     'hyper sharpness, perfect without deformations.',
                               lines=3)
         a_prompt_hint = gr.HTML("You can use a <a href='"'https://huggingface.co/spaces/MaziyarPanahi/llava-llama-3-8b'"'>LlaVa space</a> to auto-generate the description of your image.")
+        upscale = gr.Radio([["x1", 1], ["x2", 2], ["x3", 3], ["x4", 4], ["x5", 5], ["x6", 6], ["x7", 7], ["x8", 8]], label="Upscale factor", info="Resolution x1 to x8", value=2, interactive=True)
         output_format = gr.Radio([["*.png", "png"], ["*.webp", "webp"], ["*.jpeg", "jpeg"], ["*.gif", "gif"], ["*.bmp", "bmp"]], label="Image format for result", info="File extention", value="png", interactive=True)
 
     with gr.Accordion("Pre-denoising (optional)", open=False):