Update api.py

api.py

@@ -9,8 +9,9 @@ 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, FluxControlNetModel, FluxImg2ImgPipeline, FluxInpaintPipeline, CogVideoXImageToVideoPipeline
+from diffusers import FluxPipeline, FluxImg2ImgPipeline, FluxInpaintPipeline, CogVideoXImageToVideoPipeline
 from diffusers.pipelines import FluxControlNetPipeline, FluxControlNetInpaintPipeline
+from diffusers.model import FluxControlNetModel
 from diffusers.utils import load_image
 from PIL import Image
 from collections import defaultdict