metadata

app.py

@@ -19,13 +19,12 @@ EMBEDDINGS_DIR.mkdir(exist_ok=True)
 EMBEDDINGS_FILE = EMBEDDINGS_DIR / "embeddings.pkl"
 
 # ✅ Cargar el dataset de Hugging Face con imágenes decodificadas
-download_config = DownloadConfig(force_download=True, token=HF_TOKEN)
-dataset = load_dataset("Segizu/facial-recognition", download_config=download_config)
+dataset = load_dataset("Segizu/facial-recognition", data_files="metadata.csv", token=HF_TOKEN)
 if "train" in dataset:
     dataset = dataset["train"]
 
-# Asegurar que la columna 'image' sea del tipo imagen
-dataset = dataset.cast_column("image", HfImage())
+
+dataset = dataset["train"].cast_column("image", HfImage())
 
 # 🔄 Preprocesar imagen para Facenet
 def preprocess_image(img: Image.Image) -> np.ndarray:

metadata.py

@@ -0,0 +1,40 @@
+from huggingface_hub import HfApi
+import csv
+import os
+from pathlib import Path
+
+# 🔐 Configuración segura del token
+HF_TOKEN = os.getenv("HF_TOKEN")
+if not HF_TOKEN:
+    raise ValueError("⚠️ Por favor, configura la variable de entorno HF_TOKEN")
+
+# 🗂️ Configurar repositorio
+api = HfApi()
+repo_id = "Segizu/facial-recognition"
+
+try:
+    # ✅ Listar todos los archivos del dataset
+    files = api.list_repo_files(repo_id=repo_id, repo_type="dataset", token=HF_TOKEN)
+    
+    # Filtrar imágenes .jpg
+    image_files = [f for f in files if f.lower().endswith(".jpg")]
+    
+    # Guardar metadata.csv
+    metadata_path = Path("metadata.csv")
+    with open(metadata_path, "w", newline="") as f:
+        writer = csv.writer(f)
+        writer.writerow(["image"])
+        for img in image_files:
+            writer.writerow([img])
+    
+    print(f"✅ metadata.csv generado con {len(image_files)} imágenes.")
+
+except Exception as e:
+    print(f"❌ Error: {str(e)}")
+    if "401" in str(e):
+        print("⚠️ Error de autenticación. Verifica que tu token de Hugging Face sea válido.")
+    elif "404" in str(e):
+        print("⚠️ No se encontró el repositorio. Verifica que el nombre del repositorio sea correcto.")
+    else:
+        print("⚠️ Ocurrió un error inesperado.")
+