fix ui params

app_haircolor.py

@@ -98,6 +98,7 @@ def create_demo() -> gr.Blocks:
             with gr.Column():
                 edit_prompt = gr.Textbox(lines=1, label="Edit Prompt", value=DEFAULT_EDIT_PROMPT)
                 generate_size = gr.Number(label="Generate Size", value=1024)
+                adapter_weight = gr.Slider(minimum=0, maximum=2, value=1, step=0.1, label="Adapter Weight")
                 category = gr.Textbox(label="Category", value=DEFAULT_CATEGORY, visible=False)
             with gr.Column():
                 num_steps = gr.Slider(minimum=1, maximum=100, value=30, step=1, label="Num Steps")
@@ -124,7 +125,7 @@ def create_demo() -> gr.Blocks:
             outputs=[origin_area_image, croper],
         ).success(
             fn=image_to_image,
-            inputs=[origin_area_image, edit_prompt,seed, num_steps, guidance_scale],
+            inputs=[origin_area_image, edit_prompt,seed, num_steps, guidance_scale, generate_size, adapter_weight],
             outputs=[generated_image, generated_cost],
         ).success(
             fn=restore_result,