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)