Update app.py

app.py

@@ -1,3 +1,4 @@
+
 import torch
 from diffusers.utils import load_image
 from diffusers import FluxControlNetPipeline, FluxControlNetModel, FluxMultiControlNetModel
@@ -24,18 +25,18 @@ control_modes = [
 ]
 
 @spaces.GPU
-def generate_image(prompt, control_image_depth, control_mode_depth_index, use_depth, control_image_canny, control_mode_canny_index):
+def generate_image(prompt, control_image_depth, control_mode_depth, use_depth, control_image_canny, control_mode_canny):
     control_images = []
     control_modes = []
     conditioning_scales = []
 
     if use_depth:
         control_images.append(control_image_depth)
-        control_modes.append(control_mode_depth_index)
+        control_modes.append(control_modes.index(control_mode_depth))
         conditioning_scales.append(0.2)
 
     control_images.append(control_image_canny)
-    control_modes.append(control_mode_canny_index)
+    control_modes.append(control_modes.index(control_mode_canny))
     conditioning_scales.append(0.4)
 
     width, height = control_image_canny.size