Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import torch
 from diffusers.utils import load_image
 from diffusers import FluxControlNetPipeline, FluxControlNetModel, FluxMultiControlNetModel
@@ -25,18 +24,18 @@ control_modes = [
 ]
 
 @spaces.GPU
-def generate_image(prompt, control_image_depth, control_mode_depth, use_depth, control_image_canny, control_mode_canny):
+def generate_image(prompt, control_image_depth, control_mode_depth_index, use_depth, control_image_canny, control_mode_canny_index):
     control_images = []
     control_modes = []
     conditioning_scales = []
 
     if use_depth:
         control_images.append(control_image_depth)
-        control_modes.append(control_modes.index(control_mode_depth))
+        control_modes.append(control_mode_depth_index)
         conditioning_scales.append(0.2)
 
     control_images.append(control_image_canny)
-    control_modes.append(control_modes.index(control_mode_canny))
+    control_modes.append(control_mode_canny_index)
     conditioning_scales.append(0.4)
 
     width, height = control_image_canny.size
@@ -60,10 +59,10 @@ iface = gr.Interface(
     inputs=[
         gr.Text(label="Prompt"),
         gr.Image(label="Control Image (Depth)"),
-        gr.Dropdown(choices=control_modes, value="depth", label="Control Mode (Depth)"),
+        gr.Dropdown(choices=list(range(len(control_modes))), value=control_modes.index("depth"), label="Control Mode (Depth)"),
         gr.Checkbox(label="Use Depth Control Image", value=True),
         gr.Image(label="Control Image (Canny)"),
-        gr.Dropdown(choices=control_modes, value="canny", label="Control Mode (Canny)")
+        gr.Dropdown(choices=list(range(len(control_modes))), value=control_modes.index("canny"), label="Control Mode (Canny)")
     ],
     outputs=gr.Image(label="Generated Image"),
     title="FluxControlNet Image Generation",