add lineart and canny scale

app_haircolor.py

@@ -119,7 +119,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")
+                seed = gr.Number(label="Seed", value=8)
                 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")
@@ -127,7 +127,8 @@ def create_demo() -> gr.Blocks:
                 mask_expansion = gr.Number(label="Mask Expansion", value=50, visible=True)
             with gr.Column():
                 mask_dilation = gr.Slider(minimum=0, maximum=10, value=2, step=1, label="Mask Dilation")
-                seed = gr.Number(label="Seed", value=8)
+                lineart_scale = gr.Slider(minimum=0, maximum=2, value=1, step=0.1, label="Lineart Scale")
+                canny_scale = gr.Slider(minimum=0, maximum=2, value=0.5, step=0.1, label="Canny Scale")
                 g_btn = gr.Button("Edit Image")
                 
         with gr.Row():
@@ -146,7 +147,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, generate_size, adapter_weight],
+            inputs=[origin_area_image, edit_prompt,seed, num_steps, guidance_scale, generate_size, lineart_scale, canny_scale],
             outputs=[generated_image, generated_cost],
         ).success(
             fn=restore_result,