Update train_model.py

train_model.py

@@ -11,7 +11,8 @@ data = {
     'face_shape': ['Oval', 'Round', 'Square'] * 50,
     'skin_tone': ['Fair', 'Medium', 'Dark'] * 50,
     'face_size': ['Small', 'Medium', 'Large'] * 50,
-    'mask_style': ['Glitter', 'Animal', 'Floral'] * 50
+    'mask_style': ['Glitter', 'Animal', 'Floral'] * 50,
+    'mask_image': ['glitter_mask.png', 'animal_mask.png', 'floral_mask.png'] * 50  # Add paths
 }
 df = pd.DataFrame(data)