sasha HF staff commited on
Commit
ff1492f
1 Parent(s): c089fdd

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -15,8 +15,8 @@ ds.add_faiss_index(column='embeddings')
15
 
16
 
17
  def query(image, number_to_retrieve=1):
18
- prompt = model.encode(image)
19
- scores, retrieved_examples = ds.get_nearest_examples('embeddings', image, k=number_to_retrieve)
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