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

modify secret key

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