Update app.py

app.py

@@ -12,7 +12,7 @@ ds.add_faiss_index(column='embeddings')
 
 def query(image, number_to_retrieve=1):
     prompt = model.encode(image)
-    scores, retrieved_examples = ds.get_nearest_examples('embeddings', prompt, k=number_to_retrieve)
+    scores, retrieved_examples = ds.get_nearest_examples('embeddings', image, k=number_to_retrieve)
     return retrieved_examples