ranchaya commited on
Commit
b7ad995
·
1 Parent(s): 3115bc6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -6,7 +6,6 @@ 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
- import os
10
  def main():
11
  pdf_folder_path = "./PDFfiles"
12
  loader = PyPDFDirectoryLoader(pdf_folder_path)
@@ -15,7 +14,7 @@ def main():
15
  texts = text_splitter.split_documents(docs)
16
  embeddings = HuggingFaceEmbeddings()
17
  vectordb = Chroma.from_documents(documents=texts,embedding=embeddings)
18
- llm = HuggingFaceHub(repo_id="google/flan-t5-large", model_kwargs={"temperature":0,"max_length":200}, huggingfacehub_api_token= os.environ["API_KEY"]) # type: ignore
19
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff",retriever=vectordb.as_retriever(search_type="mmr", search_kwargs={'fetch_k': 30}), return_source_documents=True)
20
  query = st.text_input("Ask a question: ")
21
  qa(inputs=query,return_only_outputs=True)
 
6
  from langchain.chains.question_answering import load_qa_chain
7
  import streamlit as st
8
  from langchain.chains import RetrievalQA
 
9
  def main():
10
  pdf_folder_path = "./PDFfiles"
11
  loader = PyPDFDirectoryLoader(pdf_folder_path)
 
14
  texts = text_splitter.split_documents(docs)
15
  embeddings = HuggingFaceEmbeddings()
16
  vectordb = Chroma.from_documents(documents=texts,embedding=embeddings)
17
+ llm = HuggingFaceHub(repo_id="google/flan-t5-large", model_kwargs={"temperature":0,"max_length":200}, huggingfacehub_api_token="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") # type: ignore
18
  qa = RetrievalQA.from_chain_type(llm=llm, chain_type="stuff",retriever=vectordb.as_retriever(search_type="mmr", search_kwargs={'fetch_k': 30}), return_source_documents=True)
19
  query = st.text_input("Ask a question: ")
20
  qa(inputs=query,return_only_outputs=True)