T

app.py

@@ -120,7 +120,7 @@ def answer(system: str, context: str, question: str, user_id="demo", history="No
         context_list = [context]
         # 1.  Retrieve top-k similar passages
         if history == "Some":
-            q_vec  = embed(question).cpu()
+            q_vec  = embed(question).T.cpu()
             store  = kb[user_id]
             sims   = torch.matmul(store["vecs"], q_vec)          # [N]
             k      = min(4, sims.numel())