Update app.py

app.py

@@ -4,7 +4,7 @@ Original code by Zenafey
 """
 import gradio as gr
 
-from engine import generate_sd, generate_sdxl, transform_sd, controlnet_sd,  get_models
+from engine import generate_sd, generate_sdxl, transform_sd, controlnet_sd, upscale, get_models
 from const import CMODELS, CMODULES, SAMPLER_LIST, SDXL_MODEL_LIST
 
 
@@ -166,4 +166,6 @@ with gr.Blocks() as demo:
             with gr.Column():
                 image_output = gr.Image()
 
+        upscale_btn.click(upscale, inputs=[image_input, scale_by], outputs=image_output)
+
 demo.launch(show_api=False)