ranchaya commited on
Commit
3c1b268
·
1 Parent(s): 401605a

modify secret key

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -7,14 +7,13 @@ from langchain.chains.question_answering import load_qa_chain
7
  import streamlit as st
8
  from langchain.chains import RetrievalQA
9
  import os
10
- from dotenv import load_dotenv
11
  query = st.text_input("Ask a question: ")
12
  @st.cache_resource
13
  def llm():
14
- return HuggingFaceHub(repo_id="google/flan-t5-small", model_kwargs={"temperature":0,"max_length":200}, huggingfacehub_api_token=os.getenv("API_TOKEN")) # type: ignore
15
  @st.cache_resource
16
  def qa(query):
17
- load_dotenv()
18
  pdf_folder_path = "./PDFfiles"
19
  loader = PyPDFDirectoryLoader(pdf_folder_path)
20
  docs = loader.load()
 
7
  import streamlit as st
8
  from langchain.chains import RetrievalQA
9
  import os
10
+ HF_TOKEN = os.environ.get("API_TOKEN")
11
  query = st.text_input("Ask a question: ")
12
  @st.cache_resource
13
  def llm():
14
+ return HuggingFaceHub(repo_id="google/flan-t5-small", model_kwargs={"temperature":0,"max_length":200}, huggingfacehub_api_token=HF_TOKEN) # type: ignore
15
  @st.cache_resource
16
  def qa(query):
 
17
  pdf_folder_path = "./PDFfiles"
18
  loader = PyPDFDirectoryLoader(pdf_folder_path)
19
  docs = loader.load()