shivangibithel commited on
Commit
5c4950c
1 Parent(s): aa90ca5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -52,8 +52,8 @@ def T2Isearch(query, k=50):
52
  # inputs = text_tokenizer([query], padding=True, return_tensors="pt")
53
  # outputs = text_model(**inputs)
54
  # query_embedding = outputs.text_embeds
55
- query_embedding = test_xt[0]
56
- query_vector = np.array([query_embedding])
57
  faiss.normalize_L2(query_vector)
58
  # text_index.nprobe = index.ntotal
59
  text_index.nprobe = 100
 
52
  # inputs = text_tokenizer([query], padding=True, return_tensors="pt")
53
  # outputs = text_model(**inputs)
54
  # query_embedding = outputs.text_embeds
55
+ query_vector = test_xt[0]
56
+ # query_vector = np.array([query_embedding])
57
  faiss.normalize_L2(query_vector)
58
  # text_index.nprobe = index.ntotal
59
  text_index.nprobe = 100