Update app.py

app.py

@@ -168,6 +168,12 @@ def generate_image(images, gender, prompt, progress=gr.Progress(track_tqdm=True)
     if not prompt:
         prompt = f"Professional portrait of a {gender.lower()}"
     
+    # Add specific keywords to ensure single person
+    prompt = f"{prompt}, single person, solo portrait, one person only, centered composition"
+    
+    # Add negative prompt to prevent multiple people
+    negative_prompt = "multiple people, group photo, crowd, double portrait, triple portrait, many faces, multiple faces, two faces, three faces, multiple views, collage, photo grid"
+    
     faceid_all_embeds = []
     first_iteration = True
     preserve_face_structure = True
@@ -187,15 +193,17 @@ def generate_image(images, gender, prompt, progress=gr.Progress(track_tqdm=True)
     average_embedding = torch.mean(torch.stack(faceid_all_embeds, dim=0), dim=0)
 
     image = ip_model_plus.generate(
-        prompt=prompt,  
+        prompt=prompt,
+        negative_prompt=negative_prompt,
         faceid_embeds=average_embedding,
         scale=likeness_strength, 
         face_image=face_image, 
         shortcut=True, 
         s_scale=face_strength, 
         width=512, 
-        height=912, 
-        num_inference_steps=100
+        height=768,  # Adjusted for better single-person portrait composition
+        num_inference_steps=100,
+        guidance_scale=7.5  # Added to enforce prompt adherence
     )
     return image