fix

app.py

@@ -43,13 +43,13 @@ def load_title_embeddings():
 
 
 def get_retrieval_results(index, input_text, top_k, tokenizer, title_df):
-    batch_dict = tokenizer(f"query: {input_text}", max_length=512, padding=True, truncation=True, return_tensors='pt')
+    batch_dict = tokenizer([f"query: {input_text}"], max_length=512, padding=True, truncation=True, return_tensors='pt')
     with torch.no_grad():
         outputs = model(**batch_dict)
         query_embeddings = average_pool(outputs.last_hidden_state, batch_dict['attention_mask'])
         query_embeddings = F.normalize(query_embeddings, p=2, dim=1)
     
-    _, ids = index.search(x=np.array([query_embeddings]), k=top_k)
+    _, ids = index.search(x=query_embeddings.detach().numpy().copy(), k=top_k)
     retrieved_titles = []
     retrieved_pids = []