Cyantist8208 commited on
Commit
e501608
·
1 Parent(s): bf671c6
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -142,7 +142,7 @@ def answer(question: str, user_id="demo", history="None"):
142
  def clear_kb(user_id="demo"):
143
  if user_id in kb:
144
  kb[user_id]["texts"].clear()
145
- kb[user_id]["vecs"] = torch.empty((0, 4096))
146
  return f"Cleared KB for user '{user_id}'."
147
  else:
148
  return f"User ID '{user_id}' not found."
 
142
  def clear_kb(user_id="demo"):
143
  if user_id in kb:
144
  kb[user_id]["texts"].clear()
145
+ kb[user_id]["vecs"] = None
146
  return f"Cleared KB for user '{user_id}'."
147
  else:
148
  return f"User ID '{user_id}' not found."