srijaydeshpande commited on
Commit
ce2da6f
·
verified ·
1 Parent(s): 26922bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -68,7 +68,11 @@ def run(bucket_folder, person_name, req_id, gender='male'):
68
  if file.lower().endswith(image_extensions)
69
  ]
70
 
71
- prompt = 'Professional LinkedIn-style realistic headshot, symmetrical full face and upper body visible including shoulders and chest, centered composition with a small space above the head, wearing a formal suit and white shirt, neutral expression, captured from a short distance, realistic skin texture, exact face preserved, plain white or gray background, sharp focus, studio lighting, high-resolution, suitable for CV or resume'
 
 
 
 
72
  neg_prompt = 'Wrong face, flaws in the eyes, flaws in the face, lowres, artifacts noise, text, deformed, partially rendered objects, deformed or partially rendered eyes, deformed eyeballs, cross-eyed, blurry'
73
  scale = 1.2
74
  n_samples = 5
 
68
  if file.lower().endswith(image_extensions)
69
  ]
70
 
71
+ if(gender.lower()=='female'):
72
+ prompt = 'Professional LinkedIn-style realistic headshot for woman named ' + person_name + ', symmetrical full face and upper body visible including shoulders and chest, centered composition with a small space above the head, wearing a smart casual, neutral expression, captured from a short distance, realistic skin texture, exact face preserved, plain white or gray background, sharp focus, studio lighting, high-resolution, suitable for CV or resume'
73
+ else:
74
+ prompt = 'Professional LinkedIn-style realistic headshot for man named ' + person_name + ', symmetrical full face and upper body visible including shoulders and chest, centered composition with a small space above the head, wearing a formal suit and white shirt, neutral expression, captured from a short distance, realistic skin texture, exact face preserved, plain white or gray background, sharp focus, studio lighting, high-resolution, suitable for CV or resume'
75
+
76
  neg_prompt = 'Wrong face, flaws in the eyes, flaws in the face, lowres, artifacts noise, text, deformed, partially rendered objects, deformed or partially rendered eyes, deformed eyeballs, cross-eyed, blurry'
77
  scale = 1.2
78
  n_samples = 5