alamshoaib134 commited on
Commit
672ab3e
·
verified ·
1 Parent(s): feb4f1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -17,6 +17,7 @@ def main(dataset,qs):
17
 
18
  def dataset_change(dataset):
19
  global vectorstore
 
20
  loader = CSVLoader(dataset.name, encoding="utf-8", csv_args={'delimiter': ','})
21
  data = loader.load()
22
  vectorstore = FAISS.from_documents(data, embeddings)
 
17
 
18
  def dataset_change(dataset):
19
  global vectorstore
20
+ embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2',model_kwargs={'device': 'cpu'})
21
  loader = CSVLoader(dataset.name, encoding="utf-8", csv_args={'delimiter': ','})
22
  data = loader.load()
23
  vectorstore = FAISS.from_documents(data, embeddings)