Deadmon commited on
Commit
9c0158e
·
verified ·
1 Parent(s): 1368e65

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -24,7 +24,10 @@ control_modes = [
24
  ]
25
 
26
  @spaces.GPU
27
- def generate_image(prompt, control_image_depth, control_mode_depth, control_image_canny, control_mode_canny):
 
 
 
28
  width, height = control_image_depth.size
29
 
30
  image = pipe(
 
24
  ]
25
 
26
  @spaces.GPU
27
+ def generate_image(prompt, control_image_depth, control_mode_depth_index, control_image_canny, control_mode_canny_index):
28
+ control_mode_depth = control_modes.index(control_mode_depth_index)
29
+ control_mode_canny = control_modes.index(control_mode_canny_index)
30
+
31
  width, height = control_image_depth.size
32
 
33
  image = pipe(