99ashutosh commited on
Commit
2da369d
·
1 Parent(s): 1e834dd
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ from similarity_finder.get_similar_images import get_similar_images
9
  def my_inference_function(image_path, age, gender, race):
10
  global df
11
  r = get_similar_images(df, image_path, int(age), gender, race)
12
- return r[0]["embedding"]
13
 
14
  gradio_interface = gradio.Interface(
15
  fn = my_inference_function,
 
9
  def my_inference_function(image_path, age, gender, race):
10
  global df
11
  r = get_similar_images(df, image_path, int(age), gender, race)
12
+ return r
13
 
14
  gradio_interface = gradio.Interface(
15
  fn = my_inference_function,