Update app.py

app.py

@@ -21,14 +21,6 @@ def generate_image(prompt, negative_prompt, num_inference_steps=30, guidance_sca
         
     else:
         model_id = "SG161222/Realistic_Vision_V6.0_B1_noVAE"
-
-    if "blowjob" in prompt.split():
-        prompt = prompt + "blowjob, hand on penis,naked male"
-        pipe.load_lora_weights(lora2)
-
-    if "caught" in prompt.split():
-        prompt = prompt + "busted, male pov, male masturbation, standing, hand on door, opening door"
-        pipe.load_lora_weights(lora3)
         
 
     text_encoder = CLIPTextModel.from_pretrained(
@@ -47,6 +39,14 @@ def generate_image(prompt, negative_prompt, num_inference_steps=30, guidance_sca
         tokenizer=tokenizer,
         vae=vae
     ).to("cuda")
+
+    if "blowjob" in prompt.split():
+        prompt = prompt + "blowjob, hand on penis,naked male"
+        pipe.load_lora_weights(lora2)
+
+    if "caught" in prompt.split():
+        prompt = prompt + "busted, male pov, male masturbation, standing, hand on door, opening door"
+        pipe.load_lora_weights(lora3)
     
     if model == "Real6.0":
         pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))