Spaces:
Sleeping
Sleeping
shivangibithel
commited on
Commit
•
5c4950c
1
Parent(s):
aa90ca5
Update app.py
Browse files
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 |
-
|
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
|