Spaces:
Runtime error
Runtime error
Samuel Schmidt
commited on
Commit
·
a240077
1
Parent(s):
cb03df9
Update src/app.py
Browse files- 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,
|
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':
|
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
|