Update app.py

app.py

@@ -132,6 +132,7 @@ def retrieve_gradio(knowledge_base: str, query: str, topk: int):
 
     doc_reps_cat = torch.stack([torch.Tensor(i) for i in doc_reps], dim=0)
 
+    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)
 
     topk_values, topk_doc_ids = torch.topk(similarities, k=topk)