Update app.py

app.py

@@ -25,6 +25,7 @@ import json
 from preprocessor import Preprocessor
 from diffusers import FluxControlNetInpaintPipeline
 from diffusers.models import FluxControlNetModel
+from diffusers import FluxControlPipeline
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
@@ -38,7 +39,8 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 base_model = "black-forest-labs/FLUX.1-dev"
 
 controlnet_model = 'black-forest-labs/FLUX.1-Canny-dev'
-controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
+#controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.bfloat16)
+controlnet = FluxControlPipeline.from_pretrained("black-forest-labs/FLUX.1-Canny-dev", torch_dtype=torch.bfloat16)
 
 
 pipe = FluxControlNetInpaintPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.bfloat16).to(device)