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,  get_models
 from const import CMODELS, CMODULES, SAMPLER_LIST, SDXL_MODEL_LIST
 
 
@@ -157,4 +157,13 @@ with gr.Blocks() as demo:
                                       negative_prompt, steps, cfg_scale, seed, sampler, width, height],
                               outputs=image_output)
 
+    with gr.Tab("/upscale"):
+        with gr.Row():
+            with gr.Column():
+                image_input = gr.Image(type='filepath')
+                scale_by = gr.Slider(label="Scale by", minimum=2, maximum=4, step=2)
+                upscale_btn = gr.Button("Upscale!")
+            with gr.Column():
+                image_output = gr.Image()
+
 demo.launch(show_api=False)