k out of range

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).T.cpu()
+            q_vec  = embed(question).view(-1).cpu()
             store  = kb[user_id]
             sims   = torch.matmul(store["vecs"], q_vec)          # [N]
             if sims.numel() == 0:
@@ -160,7 +160,7 @@ def answer(system: str, context: str, question: str, user_id="demo", history="No
 
           
     except Exception as e:
-        return f"Error in app.py: {e}"
+        return f"Error in app.py: {e}, k={k}, q_vec.shape={q_vec.shape if 'q_vec' in locals() else 'N/A'}"
     finally:
         torch.cuda.empty_cache()