Ellbendls commited on
Commit
16470b6
·
1 Parent(s): b365748

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -6,6 +6,7 @@ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
6
  from transformers import pipeline
7
  import torch
8
  import textwrap
 
9
  from langchain.document_loaders import PyPDFLoader, DirectoryLoader, PDFMinerLoader
10
  from langchain.text_splitter import RecursiveCharacterTextSplitter
11
  from langchain.embeddings import SentenceTransformerEmbeddings
@@ -15,6 +16,12 @@ from langchain.chains import RetrievalQA
15
  from constants import CHROMA_SETTINGS
16
  from streamlit_chat import message
17
 
 
 
 
 
 
 
18
  st.set_page_config(layout="wide")
19
 
20
  checkpoint = "LaMini-T5-738M"
 
6
  from transformers import pipeline
7
  import torch
8
  import textwrap
9
+ from dotenv import load_dotenv
10
  from langchain.document_loaders import PyPDFLoader, DirectoryLoader, PDFMinerLoader
11
  from langchain.text_splitter import RecursiveCharacterTextSplitter
12
  from langchain.embeddings import SentenceTransformerEmbeddings
 
16
  from constants import CHROMA_SETTINGS
17
  from streamlit_chat import message
18
 
19
+
20
+ load_dotenv()
21
+
22
+ os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.getenv('HUGGINGFACEHUB_API_TOKEN')
23
+
24
+
25
  st.set_page_config(layout="wide")
26
 
27
  checkpoint = "LaMini-T5-738M"