Segizu commited on
Commit
f5a1125
·
1 Parent(s): 2955be6

metadata v12

Browse files
Files changed (1) hide show
  1. app.py +8 -1
app.py CHANGED
@@ -15,6 +15,13 @@ EMBEDDINGS_DIR = Path("embeddings")
15
  EMBEDDINGS_DIR.mkdir(exist_ok=True)
16
  EMBEDDINGS_FILE = EMBEDDINGS_DIR / "embeddings.pkl"
17
 
 
 
 
 
 
 
 
18
  # ✅ Cargar el dataset remoto desde Hugging Face Datasets con metadata.csv
19
  dataset = load_dataset(
20
  "csv",
@@ -58,7 +65,7 @@ def build_database():
58
  print(f"⚠️ Saltando item {i + j} - URL inválida: {image_url}")
59
  continue
60
 
61
- response = requests.get(image_url, timeout=10)
62
  response.raise_for_status()
63
  img = Image.open(BytesIO(response.content)).convert("RGB")
64
 
 
15
  EMBEDDINGS_DIR.mkdir(exist_ok=True)
16
  EMBEDDINGS_FILE = EMBEDDINGS_DIR / "embeddings.pkl"
17
 
18
+ headers = {}
19
+ HF_TOKEN = os.getenv("HF_TOKEN")
20
+ if HF_TOKEN:
21
+ headers["Authorization"] = f"Bearer {HF_TOKEN}"
22
+
23
+
24
+
25
  # ✅ Cargar el dataset remoto desde Hugging Face Datasets con metadata.csv
26
  dataset = load_dataset(
27
  "csv",
 
65
  print(f"⚠️ Saltando item {i + j} - URL inválida: {image_url}")
66
  continue
67
 
68
+ response = requests.get(image_url, headers=headers, timeout=10)
69
  response.raise_for_status()
70
  img = Image.open(BytesIO(response.content)).convert("RGB")
71