shivangibithel commited on
Commit
86ba518
1 Parent(s): 6b4cc3a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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