Update app.py

app.py

@@ -5,7 +5,9 @@ import gradio as gr
 import spaces
 
 
-lora_path = "OedoSoldier/detail-tweaker-lora"
+lora1 = "OedoSoldier/detail-tweaker-lora"
+lora2 = "Profakerr/blow_job"
+lora3 = "Profakerr/caught_master"
 vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to("cuda")
 
 @spaces.GPU
@@ -20,6 +22,15 @@ 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(
         model_id,
         subfolder="text_encoder"
@@ -40,7 +51,7 @@ def generate_image(prompt, negative_prompt, num_inference_steps=30, guidance_sca
     if model == "Real6.0":
         pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
 
-    pipe.load_lora_weights(lora_path)
+    
 
     pipe.scheduler = DPMSolverMultistepScheduler.from_config(
         pipe.scheduler.config,