Update app.py

app.py

@@ -8,12 +8,12 @@ from huggingface_hub import login
 import os
 from diffusers.models import AutoencoderKL
 
-vae = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae")
+vae = AutoencoderKL.from_pretrained("stabilityai/sdxl-vae", safety_checker=None)
 
 login(token=os.environ.get('HF_KEY'))
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", torch_dtype=torch.float16, vae=vae) if torch.cuda.is_available() else DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", vae=vae)
+pipe = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", torch_dtype=torch.float16, vae=vae, safety_checker=None) if torch.cuda.is_available() else DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-xl-refiner-0.9", vae=vae, safety_checker=None)
 pipe = pipe.to(device)
 
 def resize(value,img):