demoPOC commited on
Commit
ffa09c7
·
1 Parent(s): ab68327

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -33,7 +33,7 @@ from langchain.text_splitter import RecursiveCharacterTextSplitter
33
 
34
  from langchain.chains import VectorDBQA
35
 
36
- from langchain.document_loaders import UnstructuredFileLoader
37
  from langchain import PromptTemplate
38
 
39
  from langchain.chains import RetrievalQA
@@ -98,7 +98,7 @@ def loadKB(fileprovided, urlProvided, uploads_dir, request):
98
  loader = PyPDFLoader(os.path.join(uploads_dir, secure_filename(file.filename)))
99
  documents.extend(loader.load())
100
  else:
101
- loader = PyPDFLoader('./KnowledgeBase/Jio.pdf')
102
  documents.extend(loader.load())
103
 
104
  if urlProvided:
 
33
 
34
  from langchain.chains import VectorDBQA
35
 
36
+ from langchain.document_loaders import UnstructuredFileLoader, TextLoader
37
  from langchain import PromptTemplate
38
 
39
  from langchain.chains import RetrievalQA
 
98
  loader = PyPDFLoader(os.path.join(uploads_dir, secure_filename(file.filename)))
99
  documents.extend(loader.load())
100
  else:
101
+ loader = TextLoader('Jio.txt')
102
  documents.extend(loader.load())
103
 
104
  if urlProvided: