Pradeep Kumar commited on
Commit
8aaa55b
·
verified ·
1 Parent(s): 2ab33a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -9,6 +9,11 @@ from langchain.embeddings import HuggingFaceEmbeddings
9
  from langchain.vectorstores import FAISS
10
  from langchain.chains import RetrievalQA
11
  from streamlit_pdf_viewer import pdf_viewer
 
 
 
 
 
12
 
13
  def init_page() -> None:
14
  st.set_page_config(page_title="PDF Chatbot")
@@ -33,7 +38,6 @@ def main() -> None:
33
  # Sidebar: LLM selection and PDF file uploader
34
  with st.sidebar:
35
  st.title("Options")
36
- token = st.text_input("HuggingFace Token")
37
  selected_model = st.selectbox(
38
  "Select LLM",
39
  options=[
 
9
  from langchain.vectorstores import FAISS
10
  from langchain.chains import RetrievalQA
11
  from streamlit_pdf_viewer import pdf_viewer
12
+ import dotenv
13
+ dotenv.load_dotenv()
14
+ import os
15
+
16
+ token = os.getenv("Token")
17
 
18
  def init_page() -> None:
19
  st.set_page_config(page_title="PDF Chatbot")
 
38
  # Sidebar: LLM selection and PDF file uploader
39
  with st.sidebar:
40
  st.title("Options")
 
41
  selected_model = st.selectbox(
42
  "Select LLM",
43
  options=[