Update app.py

app.py

@@ -132,7 +132,7 @@ def retrieve_gradio(knowledge_base: str, query: str, topk: int):
 
     query_md5 = hashlib.md5(query.encode()).hexdigest()
 
-    doc_reps_cat = torch.stack([torch.Tensor(i) for i in doc_reps], dim=0)
+    doc_reps_cat = torch.cat([torch.Tensor(i) for i in doc_reps], dim=0)
     print(f"doc_reps_shape: {doc_reps.shape}")
     print(f"query_rep_shape: {query_rep.shape}, doc_reps_cat_shape: {doc_reps_cat.shape}")
     similarities = torch.matmul(query_rep, doc_reps_cat.T)