supertskone commited on
Commit
729713e
1 Parent(s): 92ff33f

Update load_data.py

Browse files
Files changed (1) hide show
  1. load_data.py +3 -0
load_data.py CHANGED
@@ -1,5 +1,8 @@
 
1
  from app.vectorizer import Vectorizer
2
 
 
 
3
  if __name__ == "__main__":
4
  vectorizer = Vectorizer()
5
  vectorizer.store_from_dataset(store_data=True) # Run this once to load the dataset into Pinecone
 
1
+ import os
2
  from app.vectorizer import Vectorizer
3
 
4
+ os.environ['HF_HOME'] = '/app/cache/hub'
5
+
6
  if __name__ == "__main__":
7
  vectorizer = Vectorizer()
8
  vectorizer.store_from_dataset(store_data=True) # Run this once to load the dataset into Pinecone