result_slider = ImageSlider(label='Comparator', show_label=False, elem_id="slider1", show_download_button = False)

app.py

@@ -316,7 +316,7 @@ def restore(
 
     if torch.cuda.device_count() == 0:
         gr.Warning('Set this space to GPU config to make it work.')
-        return [input_image] * 2, [input_image] * 2, None, None
+        return [input_image] * 2, [input_image], None, None
     torch.cuda.set_device(SUPIR_device)
     event_id = str(time.time_ns())
     event_dict = {'event_id': event_id, 'localtime': time.ctime(), 'prompt': prompt, 'a_prompt': a_prompt,
@@ -555,8 +555,8 @@ with gr.Blocks() 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='Comparator', show_label=False, elem_id="slider1", show_download_button = False)
+        result_gallery = gr.Gallery(label='Downloadable results', show_label=True, elem_id="gallery1")
 
     gr.Examples(
         run_on_click = True,