Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -11,6 +11,7 @@ MAX_IMAGE_SIZE = 1024
|
|
11 |
IP_ADAPTER = 'h94/IP-Adapter'
|
12 |
WEIGHT_NAME = "ip-adapter_sd15.bin"
|
13 |
WEIGHT_NAME_plus = "ip-adapter-plus_sd15.bin"
|
|
|
14 |
|
15 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
16 |
model_default = "stable-diffusion-v1-5/stable-diffusion-v1-5"
|
@@ -383,7 +384,7 @@ with gr.Blocks(css=css) as demo:
|
|
383 |
choices=[
|
384 |
"pose_estimation",
|
385 |
"edge_detection",
|
386 |
-
"
|
387 |
],
|
388 |
value="pose_estimation",
|
389 |
interactive=True,
|
@@ -429,7 +430,7 @@ with gr.Blocks(css=css) as demo:
|
|
429 |
choices=[
|
430 |
"pose_estimation",
|
431 |
"edge_detection",
|
432 |
-
"
|
433 |
],
|
434 |
value="pose_estimation",
|
435 |
interactive=True,
|
|
|
11 |
IP_ADAPTER = 'h94/IP-Adapter'
|
12 |
WEIGHT_NAME = "ip-adapter_sd15.bin"
|
13 |
WEIGHT_NAME_plus = "ip-adapter-plus_sd15.bin"
|
14 |
+
WEIGHT_NAME_face = "ip-adapter-full-face_sd15.bin"
|
15 |
|
16 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
17 |
model_default = "stable-diffusion-v1-5/stable-diffusion-v1-5"
|
|
|
384 |
choices=[
|
385 |
"pose_estimation",
|
386 |
"edge_detection",
|
387 |
+
"depth_map",
|
388 |
],
|
389 |
value="pose_estimation",
|
390 |
interactive=True,
|
|
|
430 |
choices=[
|
431 |
"pose_estimation",
|
432 |
"edge_detection",
|
433 |
+
"depth_map",
|
434 |
],
|
435 |
value="pose_estimation",
|
436 |
interactive=True,
|