Spaces:
Sleeping
Sleeping
Pradeep Kumar commited on
Update app.py
Browse files
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=[
|