change app

app.py

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

app_haircolor_inpainting.py → app_haircolor_inpaint.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"
@@ -119,11 +119,11 @@ 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=15, 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():