Spaces:
Runtime error
Runtime error
Samuel Schmidt
commited on
Commit
·
c4eb027
1
Parent(s):
1b4e6a4
Update src/app.py
Browse files- src/app.py +2 -2
src/app.py
CHANGED
@@ -22,7 +22,7 @@ def index_dataset(dataset):
|
|
22 |
|
23 |
print("LBP Embeddings")
|
24 |
lbp_model = LBPImageEncoder(8,2)
|
25 |
-
dataset_with_embeddings = dataset_with_embeddings.map(lambda row: {'lbp_embeddings': lbp_model.
|
26 |
|
27 |
# Add index
|
28 |
dataset_with_embeddings.add_faiss_index(column='color_embeddings')
|
@@ -78,7 +78,7 @@ def get_neighbors(query_image, selected_descriptor, top_k=5):
|
|
78 |
return images
|
79 |
if "LBP" in selected_descriptor:
|
80 |
lbp_model = LBPImageEncoder(8,2)
|
81 |
-
qi_embedding = lbp_model.
|
82 |
scores, retrieved_examples = dataset_with_embeddings.get_nearest_examples(
|
83 |
'lbp_embeddings', qi_embedding, k=top_k)
|
84 |
images = retrieved_examples['image']
|
|
|
22 |
|
23 |
print("LBP Embeddings")
|
24 |
lbp_model = LBPImageEncoder(8,2)
|
25 |
+
dataset_with_embeddings = dataset_with_embeddings.map(lambda row: {'lbp_embeddings': lbp_model.describe(row["image"])})
|
26 |
|
27 |
# Add index
|
28 |
dataset_with_embeddings.add_faiss_index(column='color_embeddings')
|
|
|
78 |
return images
|
79 |
if "LBP" in selected_descriptor:
|
80 |
lbp_model = LBPImageEncoder(8,2)
|
81 |
+
qi_embedding = lbp_model.describe(query_image)
|
82 |
scores, retrieved_examples = dataset_with_embeddings.get_nearest_examples(
|
83 |
'lbp_embeddings', qi_embedding, k=top_k)
|
84 |
images = retrieved_examples['image']
|