Prajith04 commited on
Commit
8d72876
·
verified ·
1 Parent(s): a8efd9f

Update vectordb_utils.py

Browse files
Files changed (1) hide show
  1. vectordb_utils.py +2 -2
vectordb_utils.py CHANGED
@@ -47,8 +47,8 @@ def populate_vectordb_from_hf():
47
 
48
  print("Populating vector DB...")
49
  for item in dataset:
50
- query = item.get("input", "").strip()
51
- response = item.get("output", "").strip()
52
  if query and response:
53
  add_to_vectordb(query, response)
54
 
 
47
 
48
  print("Populating vector DB...")
49
  for item in dataset:
50
+ query = item.get("body", "").strip()
51
+ response = item.get("answer", "").strip()
52
  if query and response:
53
  add_to_vectordb(query, response)
54