Update app.py

app.py

@@ -14,7 +14,7 @@ from PIL import Image
 from io import BytesIO
 from diffusers import StableDiffusionImg2ImgPipeline, AutoencoderKL, DiffusionPipeline
 from diffusers.utils import load_image
-from diffusers.pipelines.stable_diffusion import StableDiffusionSafetyChecker
+from safety_checker import StableDiffusionSafetyChecker
 
 DESCRIPTION = "# SDXL"
 if not torch.cuda.is_available():
@@ -64,8 +64,9 @@ def generate(
 ):
     if torch.cuda.is_available():
 
-        if not use_vae:  
-            pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model, torch_dtype=torch.float16, safety_checker:StableDiffusionSafetyChecker)
+        if not use_vae:
+            safety_checker = StableDiffusionSafetyChecker.from_pretrained("CompVis/stable-diffusion-safety-checker")
+            pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model, torch_dtype=torch.float16)
             
         if use_vae:
             vae = AutoencoderKL.from_pretrained(vaecall, torch_dtype=torch.float16)