Update api.py

api.py

@@ -9,9 +9,8 @@ import datetime
 from fastapi import FastAPI, HTTPException, Request, Response
 from fastapi.middleware.cors import CORSMiddleware
 from pydantic import BaseModel, constr, conint
-from diffusers import (FluxPipeline, FluxControlNetPipeline, 
-                        FluxControlNetModel, FluxImg2ImgPipeline, 
-                        FluxInpaintPipeline, CogVideoXImageToVideoPipeline)
+from diffusers import FluxPipeline, FluxControlNetModel, FluxImg2ImgPipeline, FluxInpaintPipeline, CogVideoXImageToVideoPipeline
+from diffusers.pipelines import FluxControlNetPipeline, FluxControlNetInpaintPipeline
 from diffusers.utils import load_image
 from PIL import Image
 from collections import defaultdict
@@ -151,7 +150,7 @@ async def set_controlnet_adapter(adapter: str, is_inpainting: bool = False):
 
     controlnet_model_path = adapter_controlnet_mapping[adapter]
     controlnet = FluxControlNetModel.from_pretrained(controlnet_model_path, torch_dtype=torch.bfloat16)
-    pipeline_cls = FluxControlNetPipeline if not is_inpainting else FluxInpaintPipeline
+    pipeline_cls = FluxControlNetPipeline if not is_inpainting else FluxControlNetInpaintPipeline
     flux_controlnet_pipe = pipeline_cls.from_pretrained(
         "pranavajay/flow", controlnet=controlnet, torch_dtype=torch.bfloat16
     )