Profakerr commited on
Commit
7410ac3
·
verified ·
1 Parent(s): 7f76025

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -9
app.py CHANGED
@@ -6,8 +6,7 @@ import spaces
6
 
7
 
8
  lora1 = "OedoSoldier/detail-tweaker-lora"
9
- lora2 = "Profakerr/blow_job"
10
- lora3 = "Profakerr/caught_master"
11
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to("cuda")
12
 
13
  @spaces.GPU
@@ -40,13 +39,7 @@ def generate_image(prompt, negative_prompt, num_inference_steps=30, guidance_sca
40
  vae=vae
41
  ).to("cuda")
42
 
43
- if "blowjob" in prompt.split():
44
- prompt = prompt + "blowjob, hand on penis,naked male"
45
- pipe.load_lora_weights(lora2)
46
-
47
- if "caught" in prompt.split():
48
- prompt = prompt + "busted, male pov, male masturbation, standing, hand on door, opening door"
49
- pipe.load_lora_weights(lora3)
50
 
51
  if model == "Real6.0":
52
  pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))
 
6
 
7
 
8
  lora1 = "OedoSoldier/detail-tweaker-lora"
9
+
 
10
  vae = AutoencoderKL.from_pretrained("stabilityai/sd-vae-ft-mse").to("cuda")
11
 
12
  @spaces.GPU
 
39
  vae=vae
40
  ).to("cuda")
41
 
42
+ pipe.load_lora_weights(lora1)
 
 
 
 
 
 
43
 
44
  if model == "Real6.0":
45
  pipe.safety_checker = lambda images, **kwargs: (images, [False] * len(images))