pranavajay commited on
Commit
c9b8f71
·
verified ·
1 Parent(s): 2ee280c

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +8 -2
api.py CHANGED
@@ -9,8 +9,14 @@ import datetime
9
  from fastapi import FastAPI, HTTPException, Request, Response
10
  from fastapi.middleware.cors import CORSMiddleware
11
  from pydantic import BaseModel, constr, conint
12
- from diffusers import FluxPipeline, FluxImg2ImgPipeline, FluxInpaintPipeline, CogVideoXImageToVideoPipeline
13
- from diffusers.pipelines import FluxControlNetPipeline, FluxControlNetInpaintPipeline
 
 
 
 
 
 
14
  from diffusers.model import FluxControlNetModel
15
  from diffusers.utils import load_image
16
  from PIL import Image
 
9
  from fastapi import FastAPI, HTTPException, Request, Response
10
  from fastapi.middleware.cors import CORSMiddleware
11
  from pydantic import BaseModel, constr, conint
12
+ from diffusers import FluxPipeline
13
+ from diffusers import FluxPipelineFluxImg2ImgPipeline
14
+ from diffusers import FluxImg2ImgPipeline
15
+ from diffusers import FluxInpaintPipeline
16
+ from diffusers import FluxInpaintPipeline
17
+ from diffusers import CogVideoXImageToVideoPipeline
18
+ from diffusers.pipelines import FluxControlNetPipeline
19
+ from diffusers.pipelines import FlluxControlNetInpaintPipeline
20
  from diffusers.model import FluxControlNetModel
21
  from diffusers.utils import load_image
22
  from PIL import Image