zhiweili commited on
Commit
f77d828
1 Parent(s): 14417c8

change to sdxl

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. app_haircolor_inpaint.py +1 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
 
3
- from app_haircolor_inpaint_adapter_15 import create_demo as create_demo_haircolor
4
 
5
  with gr.Blocks(css="style.css") as demo:
6
  with gr.Tabs():
 
1
  import gradio as gr
2
 
3
+ from app_haircolor_inpaint import create_demo as create_demo_haircolor
4
 
5
  with gr.Blocks(css="style.css") as demo:
6
  with gr.Tabs():
app_haircolor_inpaint.py CHANGED
@@ -20,7 +20,7 @@ from controlnet_aux import (
20
  CannyDetector,
21
  )
22
 
23
- BASE_MODEL = "diffusers/stable-diffusion-xl-1.0-inpainting-0.1"
24
 
25
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
26
 
 
20
  CannyDetector,
21
  )
22
 
23
+ BASE_MODEL = "stabilityai/stable-diffusion-xl-base-1.0"
24
 
25
  DEVICE = "cuda" if torch.cuda.is_available() else "cpu"
26