Update generator.py

generator.py

@@ -22,18 +22,12 @@ class Generator:
         
         # 3. Generate Prompt
         if not user_prompt.strip():
-            # Only get caption if face analysis didn't fail (to avoid double loading)
-            if face_emb is not None:
+            try:
                 generated_caption = get_caption(processed_image)
                 final_prompt = f"{Config.STYLE_TRIGGER}, {generated_caption}"
-            else:
-                # If face analysis failed, we might be here, so try captioning
-                try:
-                    generated_caption = get_caption(processed_image)
-                    final_prompt = f"{Config.STYLE_TRIGGER}, {generated_caption}"
-                except Exception as e:
-                    print(f"Captioning failed: {e}, using default prompt.")
-                    final_prompt = f"{Config.STYLE_TRIGGER}, a beautiful pixel art image"
+            except Exception as e:
+                print(f"Captioning failed: {e}, using default prompt.")
+                final_prompt = f"{Config.STYLE_TRIGGER}, a beautiful pixel art image"
         else:
             final_prompt = f"{Config.STYLE_TRIGGER}, {user_prompt}"
             
@@ -53,10 +47,7 @@ class Generator:
         else:
             print("No face detected: Disabling InstantID.")
             controlnet_conditioning_scale = [0.0, 0.4, 0.4] # Disable InstantID weight
-            # --- START FIX ---
-            # Set end to 0.5 (matching the 'if' block) to avoid 0.0 >= 0.0 error
-            control_guidance_end = [0.5, 0.8, 0.8] 
-            # --- END FIX ---
+            control_guidance_end = [0.5, 0.8, 0.8] # Set end to avoid 0.0 >= 0.0 error
             self.mh.pipeline.set_ip_adapter_scale(0.0)
 
         # 6. Run Inference
@@ -75,6 +66,9 @@ class Generator:
             num_inference_steps=8, 
             guidance_scale=1.5,
             
+            # --- ADDED ---
+            clip_skip=2
+            
         ).images[0]
         
         return result