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,
|