Update app.py

app.py

@@ -98,7 +98,8 @@ def generate_image(prompt, control_image, control_mode, num_steps=50, guidance=4
         x = get_noise(1, height, width, device=torch_device, dtype=torch.bfloat16, seed=seed)
         inp_cond = prepare(t5=t5, clip=clip, img=x, prompt=prompt)
 
-        x = denoise_controlnet(model, **inp_cond, controlnet=controlnet, timesteps=timesteps, guidance=guidance, controlnet_cond=controlnet_cond, control_mode=control_modes.index(control_mode))
+        controlnet.control_mode = control_modes.index(control_mode)  # Set the control mode directly in the controlnet model
+        x = denoise_controlnet(model, **inp_cond, controlnet=controlnet, timesteps=timesteps, guidance=guidance, controlnet_cond=controlnet_cond)
         
         x = unpack(x.float(), height, width)
         x = ae.decode(x)