aifeifei798 commited on
Commit
840c09a
·
verified ·
1 Parent(s): 759ad64

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -50,6 +50,9 @@ def infer(prompt, seed=42, randomize_seed=False, width=1024, height=1024, num_in
50
  seed = random.randint(0, MAX_SEED)
51
  generator = torch.Generator().manual_seed(seed)
52
  prompt = f"feifei, real model girl in real life, {prompt}, slight smile, Master of Light and Shadow."
 
 
 
53
  image = pipe(
54
  prompt = "",
55
  prompt_2 = prompt,
 
50
  seed = random.randint(0, MAX_SEED)
51
  generator = torch.Generator().manual_seed(seed)
52
  prompt = f"feifei, real model girl in real life, {prompt}, slight smile, Master of Light and Shadow."
53
+ prompt = re.sub("young woman", "feifei" ,prompt)
54
+ prompt = re.sub("woman", "feifei" ,prompt)
55
+ prompt = re.sub("girl", "feifei" ,prompt)
56
  image = pipe(
57
  prompt = "",
58
  prompt_2 = prompt,