Segizu commited on
Commit
78e29c2
·
1 Parent(s): 36f95d9

no cache embeddings

Browse files
Files changed (1) hide show
  1. app.py +11 -11
app.py CHANGED
@@ -12,10 +12,10 @@ HF_TOKEN = os.getenv("HF_TOKEN")
12
  if not HF_TOKEN:
13
  raise ValueError("⚠️ Por favor, configura la variable de entorno HF_TOKEN para acceder al dataset privado")
14
 
15
- # 📁 Configurar directorio de caché
16
- CACHE_DIR = Path("cache")
17
- CACHE_DIR.mkdir(exist_ok=True)
18
- EMBEDDINGS_CACHE = CACHE_DIR / "embeddings.pkl"
19
 
20
  os.system("rm -rf ~/.cache/huggingface/hub/datasets--Segizu--dataset_faces")
21
 
@@ -36,10 +36,10 @@ def preprocess_image(img):
36
 
37
  # 📦 Construir base de datos de embeddings
38
  def build_database():
39
- # Intentar cargar embeddings desde caché
40
- if EMBEDDINGS_CACHE.exists():
41
- print("📂 Cargando embeddings desde caché...")
42
- with open(EMBEDDINGS_CACHE, 'rb') as f:
43
  return pickle.load(f)
44
 
45
  print("🔄 Calculando embeddings (esto puede tomar unos minutos)...")
@@ -58,9 +58,9 @@ def build_database():
58
  except Exception as e:
59
  print(f"❌ No se pudo procesar imagen {i}: {e}")
60
 
61
- # Guardar embeddings en caché
62
- print("💾 Guardando embeddings en caché...")
63
- with open(EMBEDDINGS_CACHE, 'wb') as f:
64
  pickle.dump(database, f)
65
 
66
  return database
 
12
  if not HF_TOKEN:
13
  raise ValueError("⚠️ Por favor, configura la variable de entorno HF_TOKEN para acceder al dataset privado")
14
 
15
+ # 📁 Configurar directorio de embeddings
16
+ EMBEDDINGS_DIR = Path("embeddings")
17
+ EMBEDDINGS_DIR.mkdir(exist_ok=True)
18
+ EMBEDDINGS_FILE = EMBEDDINGS_DIR / "embeddings.pkl"
19
 
20
  os.system("rm -rf ~/.cache/huggingface/hub/datasets--Segizu--dataset_faces")
21
 
 
36
 
37
  # 📦 Construir base de datos de embeddings
38
  def build_database():
39
+ # Intentar cargar embeddings desde el archivo
40
+ if EMBEDDINGS_FILE.exists():
41
+ print("📂 Cargando embeddings desde el archivo...")
42
+ with open(EMBEDDINGS_FILE, 'rb') as f:
43
  return pickle.load(f)
44
 
45
  print("🔄 Calculando embeddings (esto puede tomar unos minutos)...")
 
58
  except Exception as e:
59
  print(f"❌ No se pudo procesar imagen {i}: {e}")
60
 
61
+ # Guardar embeddings en el archivo
62
+ print("💾 Guardando embeddings en el archivo...")
63
+ with open(EMBEDDINGS_FILE, 'wb') as f:
64
  pickle.dump(database, f)
65
 
66
  return database