ranchaya commited on
Commit
d18719e
·
1 Parent(s): 832c98c

modify secret key

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ import os
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=st.secrets("API_TOKEN")) # type: ignore
15
  @st.cache_resource
16
  def qa(query):
17
  pdf_folder_path = "./PDFfiles"
 
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.environ("API_TOKEN")) # type: ignore
15
  @st.cache_resource
16
  def qa(query):
17
  pdf_folder_path = "./PDFfiles"