99ashutosh commited on
Commit
0ebfd9e
·
1 Parent(s): 54b829f
similarity_finder/get_similar_images.py CHANGED
@@ -63,7 +63,7 @@ def get_sample_n_similar(t, df, sample_idx):
63
  continue
64
  return output_images[1:10]
65
 
66
- def add_to_dataframe(embedding, age, gender, race, dataframe):
67
  embedding_json = {}
68
  embedding_json['face'] = "user_image"
69
  embedding_objs = DeepFace.represent(img_path = image_path)
 
63
  continue
64
  return output_images[1:10]
65
 
66
+ 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)