Spaces:
Running
on
Zero
Running
on
Zero
zhiweili
commited on
Commit
•
f77d828
1
Parent(s):
14417c8
change to sdxl
Browse files- app.py +1 -1
- app_haircolor_inpaint.py +1 -1
app.py
CHANGED
@@ -1,6 +1,6 @@
|
|
1 |
import gradio as gr
|
2 |
|
3 |
-
from
|
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 = "
|
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 |
|