Lifeinhockey commited on
Commit
051efbb
·
verified ·
1 Parent(s): d2faec8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
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
- "scribble",
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
- "scribble",
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,