Segizu commited on
Commit
d4c09d2
·
1 Parent(s): 59cebaf

facial-recognition

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -8,7 +8,7 @@ os.system("rm -rf ~/.cache/huggingface/hub/datasets--Segizu--dataset_faces")
8
 
9
  # ✅ Cargar el dataset de Hugging Face forzando la descarga limpia
10
  download_config = DownloadConfig(force_download=True)
11
- dataset = load_dataset("Segizu/dataset_faces", download_config=download_config)
12
  if "train" in dataset:
13
  dataset = dataset["train"]
14
 
@@ -54,7 +54,7 @@ def find_similar_faces(uploaded_image):
54
  similarities.append((sim_score, name, db_img))
55
 
56
  similarities.sort(reverse=True)
57
- top_matches = similarities[:5]
58
 
59
  gallery_items = []
60
  text_summary = ""
 
8
 
9
  # ✅ Cargar el dataset de Hugging Face forzando la descarga limpia
10
  download_config = DownloadConfig(force_download=True)
11
+ dataset = load_dataset("Segizu/facial-recognition", download_config=download_config)
12
  if "train" in dataset:
13
  dataset = dataset["train"]
14
 
 
54
  similarities.append((sim_score, name, db_img))
55
 
56
  similarities.sort(reverse=True)
57
+ top_matches = similarities[:]
58
 
59
  gallery_items = []
60
  text_summary = ""