Samuel Schmidt commited on
Commit
a240077
·
1 Parent(s): cb03df9

Update src/app.py

Browse files
Files changed (1) hide show
  1. src/app.py +2 -2
src/app.py CHANGED
@@ -1,7 +1,7 @@
1
  from colordescriptor import ColorDescriptor
2
  from CLIP import CLIPImageEncoder
3
  from LBP import LBPImageEncoder
4
- from helper import chi2_distance, euclidian_distance
5
  import gradio as gr
6
  import os
7
  import cv2
@@ -77,7 +77,7 @@ def get_neighbors(query_image, selected_descriptor, selected_distance, top_k=5):
77
  tmp_dataset = dataset_with_embeddings.map(lambda row: {'distance': chi2_distance(histA=query_vector, histB=row['color_embeddings'])})
78
  retrieved_examples = tmp_dataset.sort("distance")[:5]
79
  else:
80
- tmp_dataset = dataset_with_embeddings.map(lambda row: {'distance': euclidian_distance(histA=query_vector, histB=row['color_embeddings'])})
81
  retrieved_examples = tmp_dataset.sort("distance")[:5]
82
  images = retrieved_examples['image'] #retrieved images is a dict, with images and embeddings
83
  return images
 
1
  from colordescriptor import ColorDescriptor
2
  from CLIP import CLIPImageEncoder
3
  from LBP import LBPImageEncoder
4
+ from helper import chi2_distance, euclidean_distance
5
  import gradio as gr
6
  import os
7
  import cv2
 
77
  tmp_dataset = dataset_with_embeddings.map(lambda row: {'distance': chi2_distance(histA=query_vector, histB=row['color_embeddings'])})
78
  retrieved_examples = tmp_dataset.sort("distance")[:5]
79
  else:
80
+ tmp_dataset = dataset_with_embeddings.map(lambda row: {'distance': euclidean_distance(histA=query_vector, histB=row['color_embeddings'])})
81
  retrieved_examples = tmp_dataset.sort("distance")[:5]
82
  images = retrieved_examples['image'] #retrieved images is a dict, with images and embeddings
83
  return images