Update app.py

app.py

@@ -30,14 +30,12 @@ def generate_image(prompt, control_image_depth, control_mode_depth_index, use_de
     conditioning_scales = []
 
     if use_depth:
-        control_mode_depth = control_modes.index(control_mode_depth_index)
         control_images.append(control_image_depth)
-        control_modes.append(control_mode_depth)
+        control_modes.append(control_mode_depth_index)
         conditioning_scales.append(0.2)
 
-    control_mode_canny = control_modes.index(control_mode_canny_index)
     control_images.append(control_image_canny)
-    control_modes.append(control_mode_canny)
+    control_modes.append(control_mode_canny_index)
     conditioning_scales.append(0.4)
 
     width, height = control_image_canny.size
@@ -61,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=2, 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=0, label="Control Mode (Canny)")
     ],
     outputs=gr.Image(label="Generated Image"),
     title="FluxControlNet Image Generation",