sooolee commited on
Commit
57ca393
1 Parent(s): 2300ef0

Update ingest.py to generate db only

Browse files
Files changed (1) hide show
  1. ingest.py +5 -3
ingest.py CHANGED
@@ -65,10 +65,12 @@ def ingest_data():
65
  embeddings = OpenAIEmbeddings()
66
  db = FAISS.from_documents(data, embeddings)
67
 
68
- # Save vectorstore
69
 
70
- with open("db.pkl", "wb") as f:
71
- pickle.dump(db, f)
 
 
72
 
73
  if __name__ == "__main__":
74
  ingest_data()
 
65
  embeddings = OpenAIEmbeddings()
66
  db = FAISS.from_documents(data, embeddings)
67
 
68
+ return db
69
 
70
+ # # Save vectorstore
71
+
72
+ # with open("db.pkl", "wb") as f:
73
+ # pickle.dump(db, f)
74
 
75
  if __name__ == "__main__":
76
  ingest_data()