Title & button

gradio_demo.py

@@ -178,7 +178,7 @@ def stage2_process(input_image, prompt, a_prompt, n_prompt, num_samples, upscale
         for i, result in enumerate(results):
             Image.fromarray(result).save(f'./history/{event_id[:5]}/{event_id[5:]}/HQ_{i}.png')
     print('End stage2_process')
-    return [input_image] + results, event_id
+    return [input_image] + results, [input_image] + results, event_id
 
 def load_and_reset(param_setting):
     print('Start load_and_reset')
@@ -227,22 +227,14 @@ def submit_feedback(event_id, fb_score, fb_text):
     else:
         return 'Submit failed, the server is not set to log history.'
 
-if torch.cuda.device_count() == 0:
-    title_md = """
-    <h1><center>SUPIR: Practicing Model Scaling for Photo-Realistic Image Restoration</center></h1>
-    
-    <p style="background-color: red;"><big><big><big><b>⚠️To use SUPIR, <a href="https://huggingface.co/spaces/Fabrice-TIERCELIN/SUPIR?duplicate=true">Duplicate this space</a> and set a GPU with 30 GB VRAM.</b>
-    
-    You can't use SUPIR directly here because this space runs on a CPU, which is not enough for SUPIR. This is a template space. Please provide feedback if you have issues.
-    </big></big></big></p>
-    """
-else:
-    title_md = """
-    <h1><center>SUPIR Image Upscaler</center></h1>
+title_html = """
+    <h1><center>SUPIR</center></h1>
+    <center>Upscale your images up to x8 freely, without account, without watermark and download it</center>
+    <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 available in this demo. If you want to auto-generate the description of your image, use another <a href="https://huggingface.co/spaces/MaziyarPanahi/llava-llama-3-8b">LLaVa space</a>.
     
-    <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></p>
+    <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>
     """
 
 
@@ -258,8 +250,16 @@ The service is a research preview intended for non-commercial use only, subject
 
 # Gradio interface
 with gr.Blocks(title="SUPIR") as interface:
+    if torch.cuda.device_count() == 0:
+        with gr.Row():
+            gr.HTML("""
+    <p style="background-color: red;"><big><big><big><b>⚠️To use SUPIR, <a href="https://huggingface.co/spaces/Fabrice-TIERCELIN/SUPIR?duplicate=true">Duplicate this space</a> and set a GPU with 30 GB VRAM.</b>
+    
+    You can't use SUPIR directly here because this space runs on a CPU, which is not enough for SUPIR. This is a template space. Please provide feedback if you have issues.
+    </big></big></big></p>
+    """)
     with gr.Row():
-        gr.HTML(title_md)
+        gr.HTML(title_html)
 
     with gr.Row():
         with gr.Column():
@@ -281,9 +281,9 @@ with gr.Blocks(title="SUPIR") as interface:
                 qs = gr.Textbox(label="Question", info="Ask LLaVa what description you want", value="Describe the image and its style in a very detailed manner. The image is a realistic photography, not an art painting.", lines=3)
 
             with gr.Accordion("Restoring options", open=False):
-                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; I discourage to increase because the process is limited to 3 min", minimum=1, maximum=4 if not args.use_image_slider else 1
                                         , value=1, step=1)
-                upscale = gr.Slider(label="Upscale factor", info="Resolution x1, x2, x3, x4, x5, x6, x7 or x8", minimum=1, maximum=8, value=1, step=1)
+                upscale = gr.Slider(label="Upscale factor", info="Resolution x1, x2, x3, x4, x5, x6, x7 or x8", minimum=1, maximum=8, value=2, step=1)
                 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)
                 s_cfg = gr.Slider(label="Text Guidance Scale", info="lower=follow the image, higher=follow the prompt", minimum=1.0, maximum=15.0,
                                   value=default_setting.s_cfg_Quality if torch.cuda.device_count() > 0 else 1.0, step=0.1)
@@ -332,17 +332,15 @@ with gr.Blocks(title="SUPIR") as interface:
 
         with gr.Column():
             gr.Markdown("<center>Restoring Output</center>")
-            if not args.use_image_slider:
-                result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery1")
-            else:
-                result_gallery = ImageSlider(label='Output', show_label=False, elem_id="gallery1")
+            result_slider = ImageSlider(label='Output', show_label=False, elem_id="slider1")
+            result_gallery = gr.Gallery(label='Output', show_label=False, elem_id="gallery1")
             with gr.Row():
                 with gr.Column():
                     denoise_button = gr.Button(value="Pre-denoise")
                 with gr.Column(visible=False):
                     llave_button = gr.Button(value="Generate description by LlaVa (disabled)")
                 with gr.Column():
-                    diffusion_button = gr.Button(value="🚀 Restore", variant = "primary")
+                    diffusion_button = gr.Button(value="🚀 Upscale/Restore", variant = "primary")
             with gr.Row():
                 with gr.Column():
                     param_setting = gr.Radio(["Quality", "Fidelity"], interactive=True, label="Parameter setting", value="Quality")
@@ -412,6 +410,7 @@ with gr.Blocks(title="SUPIR") as interface:
         spt_linear_s_stage2,
         model_select
     ], outputs = [
+        result_slider,
         result_gallery,
         event_id
     ])