imports

app.py

@@ -1,15 +1,14 @@
 import torch
 import gradio as gr
-from diffusers.utils import load_image
-from diffusers.pipelines.flux.pipeline_flux_controlnet_inpaint import FluxControlNetInpaintPipeline
-from diffusers.models.controlnet_flux import FluxControlNetModel
+from diffusers.pipelines.flux.pipeline_flux import FluxPipeline
+from diffusers.models.controlnet import ControlNetModel
 from controlnet_aux import CannyDetector
 
 base_model = 'black-forest-labs/FLUX.1-dev'
 controlnet_model = 'YishaoAI/flux-dev-controlnet-canny-kid-clothes'
 
-controlnet = FluxControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.float16)
-pipe = FluxControlNetInpaintPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.float16)
+controlnet = ControlNetModel.from_pretrained(controlnet_model, torch_dtype=torch.float16)
+pipe = FluxPipeline.from_pretrained(base_model, controlnet=controlnet, torch_dtype=torch.float16)
 pipe.enable_model_cpu_offload()
 pipe.to("cuda")