Update AutoPipelineForInpainting model in lifespan function

main.py

@@ -3,7 +3,6 @@ from fastapi.responses import StreamingResponse
 from contextlib import asynccontextmanager
 from starlette.middleware.cors import CORSMiddleware
 
-import torch
 from PIL import Image
 from io import BytesIO
 from diffusers import (
@@ -21,7 +20,7 @@ async def lifespan(app: FastAPI):
 
     img2img = AutoPipelineForImage2Image.from_pipe(text2img).to("cpu")
 
-    inpaint = AutoPipelineForInpainting.from_pipe(text2img).to("cpu")
+    inpaint = AutoPipelineForInpainting.from_pipe(img2img).to("cpu")
 
     yield {"text2img": text2img, "img2img": img2img, "inpaint": inpaint}
 
@@ -68,7 +67,7 @@ async def image_to_image(
     init_image = Image.open(BytesIO(bytes))
     init_image = init_image.convert("RGB").resize((512, 512))
 
-    image = request.state.img2img.pipe(
+    image = request.state.img2img(
         prompt,
         image=init_image,
         num_inference_steps=2,
@@ -96,7 +95,7 @@ async def inpainting(
     mask_image = Image.open(BytesIO(bytes))
     mask_image = mask_image.convert("RGB").resize((512, 512))
 
-    image = request.state.inpaint.pipe(
+    image = request.state.inpaint(
         prompt,
         image=init_image,
         mask_image=mask_image,