Update app.py

app.py

@@ -5,7 +5,7 @@ from diffusers.models import AutoencoderKL
 
 vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse")
 pipeline = diffusers.DiffusionPipeline.from_pretrained("SG161222/Realistic_Vision_V5.1_noVAE", vae=vae).to("cuda")
-
+pipeline.safety_checker = lambda images, **kwargs: (images, False)
 
 def read_content(file_path: str) -> str:
     """read the content of target file
@@ -17,9 +17,6 @@ def read_content(file_path: str) -> str:
 
 
 def predict(prompt, negative_prompt, guidance_scale, num_inference_steps, scheduler, lora, lora_weight):
-    pipeline.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
-    
-
     scheduler_class_name = scheduler.split("-")[0]
     add_kwargs = {}
     if len(scheduler.split("-")) > 1: