Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -5,11 +5,11 @@ from langchain_community.document_loaders import PyPDFLoader
|
|
| 5 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
| 6 |
from langchain_community.vectorstores import Chroma
|
| 7 |
from langchain.chains import ConversationalRetrievalChain
|
| 8 |
-
from langchain_huggingface import HuggingFaceEmbeddings
|
| 9 |
from langchain_community.llms import HuggingFacePipeline
|
| 10 |
from langchain.chains import ConversationChain
|
| 11 |
from langchain.memory import ConversationBufferMemory
|
| 12 |
-
|
| 13 |
|
| 14 |
from pathlib import Path
|
| 15 |
import chromadb
|
|
@@ -22,7 +22,7 @@ import tqdm
|
|
| 22 |
import accelerate
|
| 23 |
import re
|
| 24 |
|
| 25 |
-
|
| 26 |
list_llm = ["mistralai/Mistral-7B-Instruct-v0.2"]
|
| 27 |
list_llm_simple = [os.path.basename(llm) for llm in list_llm]
|
| 28 |
|
|
|
|
| 5 |
from langchain.text_splitter import RecursiveCharacterTextSplitter
|
| 6 |
from langchain_community.vectorstores import Chroma
|
| 7 |
from langchain.chains import ConversationalRetrievalChain
|
| 8 |
+
from langchain_huggingface import HuggingFaceEmbeddings , HuggingFaceEndpoint
|
| 9 |
from langchain_community.llms import HuggingFacePipeline
|
| 10 |
from langchain.chains import ConversationChain
|
| 11 |
from langchain.memory import ConversationBufferMemory
|
| 12 |
+
|
| 13 |
|
| 14 |
from pathlib import Path
|
| 15 |
import chromadb
|
|
|
|
| 22 |
import accelerate
|
| 23 |
import re
|
| 24 |
|
| 25 |
+
os.environ["HUGGINGFACEHUB_API_TOKEN"] = os.getenv("HUGGINGFACEHUB_API_TOKEN")
|
| 26 |
list_llm = ["mistralai/Mistral-7B-Instruct-v0.2"]
|
| 27 |
list_llm_simple = [os.path.basename(llm) for llm in list_llm]
|
| 28 |
|