Update app.py

app.py

@@ -186,7 +186,7 @@ class LLMService:
         self.visual_encoder.to(self.vit_sd_device, dtype=self.dtype)
 
         model_id_or_path = "stablediffusionapi/realistic-vision-v51"
-        self.vae_pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path, torch_dtype=torch.float16)
+        self.vae_pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id_or_path, safety_checker=None, torch_dtype=torch.float16)
         self.vae_pipe = self.vae_pipe.to(self.vit_sd_device)
 
         self.boi_token_id = self.tokenizer.encode(BOI_TOKEN, add_special_tokens=False)[0]