Pavan178 commited on
Commit
ed62921
·
verified ·
1 Parent(s): cdd9550

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- from langchain_community.llms import HuggingFaceEndpoint
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