Update app.py

app.py

@@ -98,7 +98,7 @@ 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_mode)
+        x = denoise_controlnet(model, **inp_cond, controlnet=controlnet, timesteps=timesteps, guidance=guidance, controlnet_cond=controlnet_cond, control_mode=control_modes.index(control_mode))
         
         x = unpack(x.float(), height, width)
         x = ae.decode(x)
@@ -138,4 +138,4 @@ interface = gr.Interface(
 )
 
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch(share=True)