Spaces:
Runtime error
Runtime error
Commit
·
1e834dd
1
Parent(s):
0ebfd9e
fix4
Browse files
similarity_finder/get_similar_images.py
CHANGED
@@ -67,7 +67,7 @@ def add_to_dataframe(image_path, age, gender, race, dataframe):
|
|
67 |
embedding_json = {}
|
68 |
embedding_json['face'] = "user_image"
|
69 |
embedding_objs = DeepFace.represent(img_path = image_path)
|
70 |
-
embedding_json['embedding'] = embedding
|
71 |
embedding_json['age'] = age
|
72 |
embedding_json['gender'] = gender
|
73 |
embedding_json['race'] = race
|
|
|
67 |
embedding_json = {}
|
68 |
embedding_json['face'] = "user_image"
|
69 |
embedding_objs = DeepFace.represent(img_path = image_path)
|
70 |
+
embedding_json['embedding'] = embedding_objs[0]["embedding"]
|
71 |
embedding_json['age'] = age
|
72 |
embedding_json['gender'] = gender
|
73 |
embedding_json['race'] = race
|