fix not defined

app_upscale.py

@@ -61,6 +61,7 @@ def create_demo() -> gr.Blocks:
             with gr.Column():
                 input_image = gr.Image(label="Input Image", type="pil")
             with gr.Column():
+                restored_image = gr.Image(label="Restored Image", format="png", type="pil", interactive=False)
                 origin_area_image = gr.Image(label="Origin Area Image", format="png", type="pil", interactive=False, visible=False)
                 upscaled_image = gr.Image(label="Upscaled Image", format="png", type="pil", interactive=False)
                 download_path = gr.File(label="Download the output image", interactive=False)
@@ -80,8 +81,8 @@ def create_demo() -> gr.Blocks:
             outputs=[upscaled_image, generated_cost],
         ).success(
             fn=restore_result,
-            inputs=[croper, category, enhanced_image],
-            outputs=[upscaled_image, download_path],
+            inputs=[croper, category, upscaled_image],
+            outputs=[restored_image, download_path],
         )
 
     return demo