zhiweili commited on
Commit
5e90935
1 Parent(s): 85a227c
Files changed (2) hide show
  1. app.py +1 -1
  2. app_haircolor_inpaint.py +2 -2
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import gradio as gr
2
 
3
- from app_haircolor_inpaint_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,11 +20,11 @@ from controlnet_aux import (
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
 
27
- DEFAULT_EDIT_PROMPT = "a woman, blue hair, high detailed"
28
  DEFAULT_NEGATIVE_PROMPT = "worst quality, normal quality, low quality, low res, blurry, text, watermark, logo, banner, extra digits, cropped, jpeg artifacts, signature, username, error, sketch ,duplicate, ugly, monochrome, horror, geometry, mutation, disgusting, poorly drawn face, bad face, fused face, ugly face, worst face, asymmetrical, unrealistic skin texture, bad proportions, out of frame, poorly drawn hands, cloned face, double face"
29
 
30
  DEFAULT_CATEGORY = "hair"
 
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
 
27
+ DEFAULT_EDIT_PROMPT = "hair to linen-blonde-hair"
28
  DEFAULT_NEGATIVE_PROMPT = "worst quality, normal quality, low quality, low res, blurry, text, watermark, logo, banner, extra digits, cropped, jpeg artifacts, signature, username, error, sketch ,duplicate, ugly, monochrome, horror, geometry, mutation, disgusting, poorly drawn face, bad face, fused face, ugly face, worst face, asymmetrical, unrealistic skin texture, bad proportions, out of frame, poorly drawn hands, cloned face, double face"
29
 
30
  DEFAULT_CATEGORY = "hair"