Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -15,8 +15,8 @@ ds.add_faiss_index(column='embeddings')
|
|
15 |
|
16 |
|
17 |
def query(image, number_to_retrieve=1):
|
18 |
-
|
19 |
-
scores, retrieved_examples = ds.get_nearest_examples('embeddings',
|
20 |
return retrieved_examples
|
21 |
|
22 |
|
|
|
15 |
|
16 |
|
17 |
def query(image, number_to_retrieve=1):
|
18 |
+
input_image = model.encode(image)
|
19 |
+
scores, retrieved_examples = ds.get_nearest_examples('embeddings', input_image, k=number_to_retrieve)
|
20 |
return retrieved_examples
|
21 |
|
22 |
|