add generate_size

app_upscale.py

@@ -58,6 +58,7 @@ def create_demo() -> gr.Blocks:
                 input_image_prompt = gr.Textbox(lines=1, label="Input Image Prompt", value=DEFAULT_SRC_PROMPT)
             with gr.Column():
                 num_inference_steps = gr.Number(label="Num Inference Steps", value=5)
+                generate_size = gr.Number(label="Generate Size", value=512, visible=False)
                 g_btn = gr.Button("Upscale Image")
                 
         with gr.Row():
@@ -70,7 +71,6 @@ def create_demo() -> gr.Blocks:
                 download_path = gr.File(label="Download the output image", interactive=False)
                 generated_cost = gr.Textbox(label="Time cost by step (ms):", visible=True, interactive=False)
                 category = gr.Textbox(label="Category", value=DEFAULT_CATEGORY, visible=False)
-                generate_size = gr.Number(label="Generate Size", value=1024, visible=False)
                 mask_expansion = gr.Number(label="Mask Expansion", value=20, visible=False)
                 mask_dilation = gr.Slider(minimum=0, maximum=10, value=2, step=1, label="Mask Dilation", visible=False)