fixed bugs

main.py

@@ -18,11 +18,11 @@ from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
 async def lifespan(app: FastAPI):
     feature_extractor = CLIPFeatureExtractor.from_pretrained(
         "openai/clip-vit-base-patch32"
-    ).to("cpu")
+    )
 
     safety_checker = StableDiffusionSafetyChecker.from_pretrained(
         "CompVis/stable-diffusion-safety-checker"
-    ).to("cpu")
+    )
 
     text2img = AutoPipelineForText2Image.from_pretrained(
         "stabilityai/sd-turbo",