on_select_result

gradio_demo.py

@@ -412,6 +412,8 @@ def load_and_reset(param_setting):
     return edm_steps, s_cfg, s_stage2, s_stage1, s_churn, s_noise, a_prompt, n_prompt, color_fix_type, linear_CFG, \
         linear_s_stage2, spt_linear_CFG, spt_linear_s_stage2, model_select
 
+def on_select_result(result_gallery, evt: gr.SelectData):
+    return [result_gallery[0], result_gallery[evt.index]]
 
 def submit_feedback(event_id, fb_score, fb_text):
     if args.log_history:
@@ -508,6 +510,7 @@ with gr.Blocks(title="SUPIR") as interface:
             with gr.Column():
                 color_fix_type = gr.Radio(["None", "AdaIn", "Wavelet"], label="Color-Fix Type", info="AdaIn=Improve following a style, Wavelet=For JPEG artifacts", value="Wavelet",
                                           interactive=True)
+        allocation = gr.Radio([["1 min", 1], ["2 min", 2], ["3 min", 3], ["4 min", 4], ["5 min", 5], ["6 min", 6], ["7 min", 7], ["8 min", 8], ["9 min", 9]], label="GPU allocation time", info="lower=May abort run, higher=Time penalty for next runs", value=6, interactive=True)
         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)
         s_stage2 = gr.Slider(label="Restoring Guidance Strength", minimum=0., maximum=1., value=1., step=0.05)
@@ -529,7 +532,6 @@ with gr.Blocks(title="SUPIR") as interface:
             with gr.Column():
                 ae_dtype = gr.Radio(['fp32', 'bf16'], label="Auto-Encoder Data Type", value="bf16",
                                     interactive=True)
-        allocation = gr.Radio([["1 min", 1], ["2 min", 2], ["3 min", 3], ["4 min", 4], ["5 min", 5], ["6 min", 6], ["7 min", 7], ["8 min", 8], ["9 min", 9]], label="GPU allocation time", info="lower=May abort run, higher=Time penalty for next runs", value=6, interactive=True)
         randomize_seed = gr.Checkbox(label = "\U0001F3B2 Randomize seed", value = True, info = "If checked, result is always different")
         seed = gr.Slider(label="Seed", minimum=0, maximum=2147483647, step=1, randomize=True)
         with gr.Group():
@@ -542,8 +544,8 @@ with gr.Blocks(title="SUPIR") as interface:
         diffusion_button = gr.Button(value="🚀 Upscale/Restore", variant = "primary", elem_id="process_button")
 
         restore_information = gr.HTML(value="Restart the process to get another result.", visible=False)
-        result_slider = ImageSlider(label='Output', show_label=True, elem_id="slider1")
-        result_gallery = gr.Gallery(label='Output', show_label=True, elem_id="gallery1")
+        result_slider = ImageSlider(label='Output', show_label=True, elem_id="slider1", value=["./Examples/Example1.png", "./Examples/Example2.jpeg"])
+        result_gallery = gr.Gallery(label='Output', show_label=True, elem_id="gallery1", value=["./Examples/Example1.png", "./Examples/Example2.jpeg"])
 
     with gr.Accordion("Feedback", open=True, visible=False):
         fb_score = gr.Slider(label="Feedback Score", minimum=1, maximum=5, value=3, step=1, interactive=True)
@@ -723,6 +725,8 @@ with gr.Blocks(title="SUPIR") as interface:
         event_id
     ])
 
+    result_gallery.select(on_select_result, result_gallery, result_slider)
+
     restart_button.click(fn = load_and_reset, inputs = [
         param_setting
     ], outputs = [