Update config.py

config.py

@@ -27,7 +27,7 @@ LORA_CHOICES = ["RetroArt", "VGA", "LucasArt", "None"]
 
 # Face detection configuration
 FACE_DETECTION_CONFIG = {
-    "model_name": "antelopev2",
+    "model_name": "pixagram_face_models", # <-- UPDATED from "antelopev2"
     "det_size": (640, 640),
     "ctx_id": 0
 }
@@ -48,7 +48,7 @@ DEFAULT_PARAMS = {
     "strength": 0.50,
     "depth_control_scale": 0.75,
     "identity_control_scale": 0.85,
-    "expression_control_scale": 0.6,  # <-- NEW
+    "expression_control_scale": 0.6,
     "lora_choice": "RetroArt",
     "lora_scale": 1.0,
     "identity_preservation": 1.2,
@@ -66,7 +66,7 @@ PRESETS = {
         "lora_scale": 0.8,
         "depth_control_scale": 0.65,
         "identity_control_scale": 0.95,
-        "expression_control_scale": 0.8,  # <-- NEW
+        "expression_control_scale": 0.8,
         "description": "Maximum face preservation - 96-98% similarity (Level 3)"
     },
     "Premium Portrait": {
@@ -76,7 +76,7 @@ PRESETS = {
         "lora_scale": 1.1,
         "depth_control_scale": 0.75,
         "identity_control_scale": 0.85,
-        "expression_control_scale": 0.6,  # <-- NEW
+        "expression_control_scale": 0.6,
         "description": "Optimized balanced - strong pixel art + excellent face (92-94% similarity)"
     },
     "Balanced Portrait": {
@@ -86,7 +86,7 @@ PRESETS = {
         "lora_scale": 1.0,
         "depth_control_scale": 0.75,
         "identity_control_scale": 0.85,
-        "expression_control_scale": 0.6,  # <-- NEW
+        "expression_control_scale": 0.6,
         "description": "Good balance between fidelity and style - 90-93% similarity"
     },
     "Artistic Excellence": {
@@ -96,7 +96,7 @@ PRESETS = {
         "lora_scale": 1.2,
         "depth_control_scale": 0.78,
         "identity_control_scale": 0.75,
-        "expression_control_scale": 0.45,  # <-- NEW
+        "expression_control_scale": 0.45,
         "description": "Creative with strong likeness - 88-91% similarity"
     },
     "Style Focus": {
@@ -106,7 +106,7 @@ PRESETS = {
         "lora_scale": 1.4,
         "depth_control_scale": 0.82,
         "identity_control_scale": 0.65,
-        "expression_control_scale": 0.3,  # <-- NEW
+        "expression_control_scale": 0.3,
         "description": "Maximum pixel art style - 83-87% similarity"
     },
     "Subtle Enhancement": {
@@ -116,7 +116,7 @@ PRESETS = {
         "lora_scale": 0.75,
         "depth_control_scale": 0.60,
         "identity_control_scale": 0.98,
-        "expression_control_scale": 0.85,  # <-- NEW
+        "expression_control_scale": 0.85,
         "description": "Minimal transformation, photo-realistic - 97-99% similarity"
     }
 }
@@ -164,24 +164,24 @@ CAPTION_CONFIG = {
 
 # Color matching settings
 COLOR_MATCH_CONFIG = {
-    "lab_lightness_blend": 0.15,  # 15% adjustment to L channel
-    "lab_color_blend_preserved": 0.05,  # 5% adjustment with saturation preservation
-    "lab_color_blend_full": 0.20,  # 20% adjustment without preservation
-    "saturation_boost": 1.05,  # Minimal saturation boost
+    "lab_lightness_blend": 0.15,
+    "lab_color_blend_preserved": 0.05,
+    "lab_color_blend_full": 0.20,
+    "saturation_boost": 1.05,
     "gaussian_blur_kernel": (51, 51),
     "gaussian_blur_sigma": 20
 }
 
 # Face mask settings
 FACE_MASK_CONFIG = {
-    "padding": 0.1,  # 10% padding around face
-    "feather": 30  # Blur radius for soft edges
+    "padding": 0.1,
+    "feather": 30
 }
 
 # Model download retry settings
 DOWNLOAD_CONFIG = {
     "max_retries": 3,
-    "retry_delay": 2  # seconds
+    "retry_delay": 2
 }
 
 # Age brackets for demographic detection