ranchaya commited on
Commit
8976c33
·
1 Parent(s): 06a07e3

modify secret key

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -10,8 +10,7 @@ import os
10
  query = st.text_input("Ask a question: ")
11
  @st.cache_resource
12
  def llm():
13
- os.environ.get()
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"
 
10
  query = st.text_input("Ask a question: ")
11
  @st.cache_resource
12
  def llm():
13
+ return HuggingFaceHub(repo_id="google/flan-t5-small", model_kwargs={"temperature":0,"max_length":200}, huggingfacehub_api_token=os.getenv("API_TOKEN")) # type: ignore
 
14
  @st.cache_resource
15
  def qa(query):
16
  pdf_folder_path = "./PDFfiles"