Update app.py

app.py

@@ -33,7 +33,7 @@ CONTROL_MODES = {
 def preprocess_image(image, target_width, target_height):
     image = image.resize((target_width, target_height), Image.LANCZOS)
     return image
-    
+
 @spaces.GPU(duration=120)
 def generate_image(prompt, control_image, control_mode, controlnet_conditioning_scale, num_steps, guidance, width, height, seed, random_seed):
     if random_seed:
@@ -49,12 +49,15 @@ def generate_image(prompt, control_image, control_mode, controlnet_conditioning_
     # Preprocess control image
     control_image = preprocess_image(control_image, width, height)
     
+    # Ensure control_mode is an integer
+    control_mode_index = int(control_mode)
+
     # Generate the image with the selected control mode and other parameters
     with torch.no_grad():
         image = pipe(
             prompt,
             control_image=control_image,
-            control_mode=control_mode,  # Pass control mode explicitly
+            control_mode=control_mode_index,  # Pass control mode as an integer
             width=width,
             height=height,
             controlnet_conditioning_scale=controlnet_conditioning_scale,
@@ -70,7 +73,7 @@ interface = gr.Interface(
     inputs=[
         gr.Textbox(label="Prompt"),
         gr.Image(type="pil", label="Control Image"),
-        gr.Dropdown(choices=[f"{i}: {name}" for i, name in CONTROL_MODES.items()], type="index", label="Control Mode", value=0),
+        gr.Dropdown(choices=[(i, name) for i, name in CONTROL_MODES.items()], label="Control Mode", value=0),  # Correct value and format for dropdown
         gr.Slider(minimum=0.1, maximum=1.0, step=0.1, value=0.5, label="ControlNet Conditioning Scale"),
         gr.Slider(step=1, minimum=1, maximum=64, value=24, label="Num Steps"),
         gr.Slider(minimum=0.1, maximum=10, value=3.5, label="Guidance"),