Spaces:
Sleeping
Sleeping
shivangibithel
commited on
Commit
•
86ba518
1
Parent(s):
6b4cc3a
Update app.py
Browse files
app.py
CHANGED
@@ -57,6 +57,7 @@ def T2Isearch(query, k=50):
|
|
57 |
outputs = text_model(**inputs)
|
58 |
query_embedding = outputs.text_embeds
|
59 |
query_vector = query_embedding.detach().numpy()
|
|
|
60 |
query_vector = query_vector.reshape(1,512)
|
61 |
faiss.normalize_L2(query_vector)
|
62 |
index.nprobe = index.ntotal
|
|
|
57 |
outputs = text_model(**inputs)
|
58 |
query_embedding = outputs.text_embeds
|
59 |
query_vector = query_embedding.detach().numpy()
|
60 |
+
# query_vector = np.concatenate((query_vector[0], query_vector[1]), dtype=np.float32)
|
61 |
query_vector = query_vector.reshape(1,512)
|
62 |
faiss.normalize_L2(query_vector)
|
63 |
index.nprobe = index.ntotal
|