Resolution

gradio_demo.py

@@ -78,7 +78,7 @@ def check(input_image):
 def reset_feedback():
     return 3, ''
 
-@spaces.GPU(duration=240)
+@spaces.GPU(duration=3600)
 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=3600)
 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=3600)
 def stage2_process(
     noisy_image,
     denoise_image,
@@ -145,6 +145,7 @@ def stage2_process(
 ):
     start = time.time()
     print('Start stage2_process')
+    print(a_prompt)
     if torch.cuda.device_count() == 0:
         gr.Warning('Set this space to GPU config to make it work.')
         return None, None, None
@@ -213,7 +214,14 @@ def stage2_process(
     secondes = secondes - (minutes * 60)
     hours = minutes // 60
     minutes = minutes - (hours * 60)
-    information = ("Restart the process to get another result. " if randomize_seed else "") + "The image(s) has(ve) been generated in " + ((str(hours) + " h, ") if hours != 0 else "") + ((str(minutes) + " min, ") if hours != 0 or minutes != 0 else "") + str(secondes) + " sec."
+    information = ("Start the process again if you want a different result. " if randomize_seed else "") + \
+    "The new image resolution is " + str(result_width) + \
+    " pixels large and " + str(result_height) + \
+    " pixels high, so a resolution of " + f'{result_width * result_height:,}' + " pixels. " + \
+    "The image(s) has(ve) been generated in " + \
+    ((str(hours) + " h, ") if hours != 0 else "") + \
+    ((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
 
@@ -271,13 +279,22 @@ title_html = """
     <big><center>Upscale your images up to x8 freely, without account, without watermark and download it</center></big>
     <br/>
     
-    <p>SUPIR is a practicing model scaling for photo-realistic image restoration. It is still a research project under tested and is not yet a stable commercial product. 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.
+    <p>This is an online demo of SUPIR, a practicing model scaling for photo-realistic image restoration.
+    It is still a research project under tested and is not yet a stable commercial product.
+    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.
     
     <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>
     """
 
 
 claim_md = """
+## **Piracy**
+
+The images are not stored but the logs are saved during a month.
+
 ## **Terms of use**
 
 By using this service, users are required to agree to the following terms: The service is a research preview intended for non-commercial use only. It only provides limited safety measures and may generate offensive content. It must not be used for any illegal, harmful, violent, racist, or sexual purposes. The service may collect user dialogue data for future research. Please submit a feedback to us if you get any inappropriate answer! We will collect those to keep improving our models. For an optimal experience, please use desktop computers for this demo, as mobile devices may compromise its quality.
@@ -309,7 +326,8 @@ with gr.Blocks(title="SUPIR") as interface:
                                 'Grading, ultra HD, extreme meticulous detailing, skin pore detailing, '
                                 '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.")
+    with gr.Group():
+        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.")
 
     with gr.Accordion("Pre-denoising (optional)", open=False):
         gamma_correction = gr.Slider(label="Gamma Correction", minimum=0.1, maximum=2.0, value=1.0, step=0.1)
@@ -331,7 +349,7 @@ with gr.Blocks(title="SUPIR") as interface:
                                     'deformed, lowres, over-smooth',
                               lines=3)
         edm_steps = gr.Slider(label="Steps", info="lower=faster, higher=more details", minimum=1, maximum=200, value=default_setting.edm_steps if torch.cuda.device_count() > 0 else 1, step=1)
-        num_samples = gr.Slider(label="Num Samples", info="Number of generated results; I discourage to increase because the process is limited to 4 min", minimum=1, maximum=4 if not args.use_image_slider else 1
+        num_samples = gr.Slider(label="Num Samples", info="Number of generated results", minimum=1, maximum=4 if not args.use_image_slider else 1
                                 , value=1, step=1)
         with gr.Row():
             with gr.Column():