0

app.py

@@ -123,6 +123,8 @@ def answer(system: str, context: str, question: str, user_id="demo", history="No
             q_vec  = embed(question).T.cpu()
             store  = kb[user_id]
             sims   = torch.matmul(store["vecs"], q_vec)          # [N]
+            if sims.numel() == 0:
+                return "Knowledge base is empty or corrupted."
             k      = min(4, sims.numel())
             idxs   = torch.topk(sims, k=k).indices.tolist()
             context_list += [store["texts"][i] for i in idxs]