Do not return the original

app.py

@@ -81,10 +81,7 @@ def check(input_image):
     if input_image is None:
         raise gr.Error("Please provide an image to restore.")
 
-def reset_feedback():
-    return 3, ''
-
-@spaces.GPU(duration=480)
+@spaces.GPU(duration=420)
 def stage1_process(input_image, gamma_correction):
     print('stage1_process ==>>')
     if torch.cuda.device_count() == 0:
@@ -106,22 +103,6 @@ def stage1_process(input_image, gamma_correction):
     print('<<== stage1_process')
     return LQ, gr.update(visible = True)
 
-@spaces.GPU(duration=540)
-def llave_process(input_image, temperature, top_p, qs=None):
-    print('llave_process ==>>')
-    if torch.cuda.device_count() == 0:
-        gr.Warning('Set this space to GPU config to make it work.')
-        return 'Set this space to GPU config to make it work.'
-    torch.cuda.set_device(LLaVA_device)
-    if use_llava:
-        LQ = HWC3(input_image)
-        LQ = Image.fromarray(LQ.astype('uint8'))
-        captions = llava_agent.gen_image_caption([LQ], temperature=temperature, top_p=top_p, qs=qs)
-    else:
-        captions = ['LLaVA is not available. Please add text manually.']
-    print('<<== llave_process')
-    return captions[0]
-
 def stage2_process(
     noisy_image,
     rotation,
@@ -413,7 +394,7 @@ def restore(
     print(information)
 
     # Only one image can be shown in the slider
-    return [input_image] + [results[0]], gr.update(format = output_format, value = [input_image] + results), gr.update(value = information, visible = True)
+    return [results[0]], gr.update(format = output_format, value = results), gr.update(value = information, visible = True)
 
 def load_and_reset(param_setting):
     print('load_and_reset ==>>')
@@ -456,22 +437,9 @@ def log_information(result_gallery):
         for i, result in enumerate(result_gallery):
             print(result[0])
 
-def on_select_result(result_gallery, evt: gr.SelectData):
+def on_select_result(result_slider, result_gallery, evt: gr.SelectData):
     print('on_select_result')
-    return [result_gallery[0][0], result_gallery[evt.index][0]]
-
-def submit_feedback(event_id, fb_score, fb_text):
-    if args.log_history:
-        with open(f'./history/{event_id[:5]}/{event_id[5:]}/logs.txt', 'r') as f:
-            event_dict = eval(f.read())
-        f.close()
-        event_dict['feedback'] = {'score': fb_score, 'text': fb_text}
-        with open(f'./history/{event_id[:5]}/{event_id[5:]}/logs.txt', 'w') as f:
-            f.write(str(event_dict))
-        f.close()
-        return 'Submit successfully, thank you for your comments!'
-    else:
-        return 'Submit failed, the server is not set to log history.'
+    return [result_slider[0], result_gallery[evt.index][0]]
 
 title_html = """
     <h1><center>SUPIR</center></h1>
@@ -503,7 +471,7 @@ The service is a research preview intended for non-commercial use only, subject
 """
 
 # Gradio interface
-with gr.Blocks(title="SUPIR") as interface:
+with gr.Blocks() as interface:
     if torch.cuda.device_count() == 0:
         with gr.Row():
             gr.HTML("""
@@ -760,7 +728,7 @@ with gr.Blocks(title="SUPIR") as interface:
         result_gallery
     ], outputs = [], queue = False, show_progress = False)
 
-    result_gallery.select(on_select_result, result_gallery, result_slider)
+    result_gallery.select(on_select_result, [result_slider, result_gallery], result_slider)
 
     restart_button.click(fn = load_and_reset, inputs = [
         param_setting