Orami01 commited on
Commit
e377924
·
1 Parent(s): dc79961

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,6 @@ DB_FAISS_PATH = 'vectorstore/db_faiss'
16
  from transformers import pipeline
17
 
18
  pipe = pipeline("text-generation",model="mistralai/Mistral-7B-v0.1",model_type="llama",max_new_tokens=512,temperature=0.1 )
19
- llm=pipe
20
 
21
 
22
 
@@ -39,6 +38,7 @@ if uploaded_file:
39
  loader = CSVLoader(file_path=tmp_file_path, encoding="utf-8", csv_args={'delimiter': ','})
40
  data = loader.load()
41
  embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2',model_kwargs={'device': 'cpu'})
 
42
  db.save_local(DB_FAISS_PATH)
43
  llm = load_llm()
44
  chain = ConversationalRetrievalChain.from_llm(llm=llm, retriever=db.as_retriever())
 
16
  from transformers import pipeline
17
 
18
  pipe = pipeline("text-generation",model="mistralai/Mistral-7B-v0.1",model_type="llama",max_new_tokens=512,temperature=0.1 )
 
19
 
20
 
21
 
 
38
  loader = CSVLoader(file_path=tmp_file_path, encoding="utf-8", csv_args={'delimiter': ','})
39
  data = loader.load()
40
  embeddings = HuggingFaceEmbeddings(model_name='sentence-transformers/all-MiniLM-L6-v2',model_kwargs={'device': 'cpu'})
41
+ db = FAISS.from_documents(data, embeddings)
42
  db.save_local(DB_FAISS_PATH)
43
  llm = load_llm()
44
  chain = ConversationalRetrievalChain.from_llm(llm=llm, retriever=db.as_retriever())