Update model/label_encoders.pkl

model/label_encoders.pkl

@@ -17,10 +17,17 @@ encoders = {
     'face_shape': LabelEncoder().fit(df['face_shape'].unique()),
     'skin_tone': LabelEncoder().fit(df['skin_tone'].unique()),
     'face_size': LabelEncoder().fit(df['face_size'].unique()),
-    'mask_style': LabelEncoder().fit(df['mask_style'].unique())
+    'mask_style': LabelEncoder().fit(df['mask_style'].unique()),
+    # Add mask image mappings
+    'mask_images': {
+        0: 'masks/glitter.png',    # Corresponds to 'Glitter' class
+        1: 'masks/animal.png',     # Corresponds to 'Animal' class
+        2: 'masks/floral.png'      # Corresponds to 'Floral' class
+    }
 }
 
 # Create model directory if not exists
+os.makedirs('masks', exist_ok=True)  # Ensure masks directory exists
 os.makedirs('model', exist_ok=True)
 
 # Save encoders dictionary
@@ -31,4 +38,9 @@ print("Label Encoders Saved Successfully!")
 print("Face Shape Classes:", encoders['face_shape'].classes_)
 print("Skin Tone Classes:", encoders['skin_tone'].classes_)
 print("Face Size Classes:", encoders['face_size'].classes_)
-print("Mask Style Classes:", encoders['mask_style'].classes_)
+print("Mask Style Classes:", encoders['mask_style'].classes_)
+print("\nMask Image Mappings:")
+for class_idx, path in encoders['mask_images'].items():
+    print(f"Class {class_idx} ({encoders['mask_style'].classes_[class_idx]}): {path}")
+    if not os.path.exists(path):
+        print(f"  → Warning: File not found at '{path}'")