ranchaya commited on
Commit
401605a
·
1 Parent(s): 9422ca1

modify secret key

Browse files
Files changed (3) hide show
  1. .gitattributes +1 -1
  2. .gitignore +1 -0
  3. app.py +4 -1
.gitattributes CHANGED
@@ -32,4 +32,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
32
  *.xz filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
- *tfevents* filter=lfs diff=lfs merge=lfs -text
 
32
  *.xz filter=lfs diff=lfs merge=lfs -text
33
  *.zip filter=lfs diff=lfs merge=lfs -text
34
  *.zst filter=lfs diff=lfs merge=lfs -text
35
+ *tfevents* filter=lfs diff=lfs merge=lfs -text
.gitignore ADDED
@@ -0,0 +1 @@
 
 
1
+ .env
app.py CHANGED
@@ -6,12 +6,15 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
6
  from langchain.chains.question_answering import load_qa_chain
7
  import streamlit as st
8
  from langchain.chains import RetrievalQA
 
 
9
  query = st.text_input("Ask a question: ")
10
  @st.cache_resource
11
  def llm():
12
- return HuggingFaceHub(repo_id="google/flan-t5-small", model_kwargs={"temperature":0,"max_length":200}, huggingfacehub_api_token="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") # type: ignore
13
  @st.cache_resource
14
  def qa(query):
 
15
  pdf_folder_path = "./PDFfiles"
16
  loader = PyPDFDirectoryLoader(pdf_folder_path)
17
  docs = loader.load()
 
6
  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()