Update app.py

app.py

@@ -52,8 +52,9 @@ def T2Isearch(query, k=50):
     inputs = text_tokenizer([query], padding=True, return_tensors="pt")    
     outputs = text_model(**inputs)
     query_embedding = outputs.text_embeds
+    query_vector = query_embedding.detach().numpy()
     # query_vector = test_xt[0]
-    query_vector = np.array([query_embedding])
+    # query_vector = np.array([query_embedding])
     faiss.normalize_L2(query_vector)
     # text_index.nprobe = index.ntotal
     text_index.nprobe = 100