Linoy Tsaban commited on
Commit
ab9ff74
·
1 Parent(s): de3ce9a

Update app.py

Browse files

added spooky concepts

Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -214,7 +214,9 @@ def edit(sdxl_loras,
214
 
215
  latnets = wts[skip].expand(1, -1, -1, -1)
216
 
217
- editing_prompt = [sdxl_loras[random_lora_index]["trigger_word"]]
 
 
218
  reverse_editing_direction = [False]
219
  edit_warmup_steps = [2]
220
  edit_guidance_scale = [sdxl_loras[random_lora_index]["edit_guidance_scale"]]
 
214
 
215
  latnets = wts[skip].expand(1, -1, -1, -1)
216
 
217
+ spooky_concept = random.choice([' spooky witch', ' spooky vampire', ' spooky werewolf', ' spooky ghost', ' spooky wizard', ' spooky pumpkin', ' spooky pirate'])
218
+ print("spooky concept is: ", spooky_concept)
219
+ editing_prompt = [sdxl_loras[random_lora_index]["trigger_word"]+ spooky_concept]
220
  reverse_editing_direction = [False]
221
  edit_warmup_steps = [2]
222
  edit_guidance_scale = [sdxl_loras[random_lora_index]["edit_guidance_scale"]]