metadata v2

app.py

@@ -23,8 +23,8 @@ dataset = load_dataset("Segizu/facial-recognition", data_files="metadata.csv", t
 if "train" in dataset:
     dataset = dataset["train"]
 
-
-dataset = dataset["train"].cast_column("image", HfImage())
+# Cast the image column to HfImage format
+dataset = dataset.cast_column("image", HfImage())
 
 # 🔄 Preprocesar imagen para Facenet
 def preprocess_image(img: Image.Image) -> np.ndarray: