Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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())
|