change app

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 
-from app_haircolor_inpaint import create_demo as create_demo_haircolor
+from app_haircolor import create_demo as create_demo_haircolor
 
 with gr.Blocks(css="style.css") as demo:
     with gr.Tabs():

app_haircolor.py

@@ -19,7 +19,7 @@ from controlnet_aux import (
     CannyDetector,
 )
 
-BASE_MODEL = "stabilityai/stable-diffusion-xl-base-1.0"
+BASE_MODEL = "SG161222/RealVisXL_V5.0_Lightning"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 DEFAULT_EDIT_PROMPT = "a woman, blue hair, high detailed"
@@ -75,9 +75,9 @@ def image_to_image(
     run_task_time = 0
     time_cost_str = ''
     run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
-    lineart_image = lineart_detector(input_image, int(generate_size*0.375), generate_size)
+    lineart_image = lineart_detector(input_image, 384, generate_size)
     run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
-    canny_image = canndy_detector(input_image, int(generate_size*0.375), generate_size)
+    canny_image = canndy_detector(input_image, 384, generate_size)
     run_task_time, time_cost_str = get_time_cost(run_task_time, time_cost_str)
 
     cond_image = [lineart_image, canny_image]
@@ -118,17 +118,17 @@ def create_demo() -> gr.Blocks:
         with gr.Row():
             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)
+                generate_size = gr.Number(label="Generate Size", value=512)
                 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")
+                num_steps = gr.Slider(minimum=1, maximum=100, value=10, step=1, label="Num Steps")
                 guidance_scale = gr.Slider(minimum=0, maximum=30, value=5, step=0.5, label="Guidance Scale")
                 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")
-                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")
+                lineart_scale = gr.Slider(minimum=0, maximum=2, value=0.3, step=0.1, label="Lineart Scale")
+                canny_scale = gr.Slider(minimum=0, maximum=2, value=0.7, step=0.1, label="Canny Scale")
                 g_btn = gr.Button("Edit Image")
                 
         with gr.Row():

app_haircolor_inpaint.py

@@ -19,8 +19,8 @@ from controlnet_aux import (
     CannyDetector,
 )
 
-# BASE_MODEL = "SG161222/RealVisXL_V5.0_Lightning"
-BASE_MODEL = "RunDiffusion/Juggernaut-XL-v9"
+BASE_MODEL = "SG161222/RealVisXL_V5.0_Lightning"
+# BASE_MODEL = "RunDiffusion/Juggernaut-XL-v9"
 DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
 
 DEFAULT_EDIT_PROMPT = "a woman, blue hair, high detailed"