99ashutosh commited on
Commit
1e834dd
·
1 Parent(s): 0ebfd9e
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