captain-awesome commited on
Commit
9097403
·
1 Parent(s): bf1769e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -5
app.py CHANGED
@@ -140,11 +140,11 @@ def create_vector_database(uploaded_files):
140
  loaded_documents = []
141
  for uploaded_file in uploaded_files:
142
  # file_extension = os.path.splitext(uploaded_file.name)[-1].lower()[1:]
143
- file_extension = os.path.splitext(uploaded_file.name)[-1][1:].lower()
144
- if file_extension in loaders:
145
- loader_cls = loaders[file_extension]
146
- loader = loader_cls()
147
- loaded_documents.extend(loader.load(uploaded_file))
148
 
149
  # all_loaders = [pdf_loader, markdown_loader, text_loader, csv_loader, python_loader, epub_loader, html_loader, ppt_loader, pptx_loader, doc_loader, docx_loader, odt_loader, notebook_loader]
150
 
 
140
  loaded_documents = []
141
  for uploaded_file in uploaded_files:
142
  # file_extension = os.path.splitext(uploaded_file.name)[-1].lower()[1:]
143
+ file_extension = os.path.splitext(uploaded_file.name)[-1][1:].lower()
144
+ if file_extension in loaders:
145
+ loader_cls = loaders[file_extension]
146
+ loader = loader_cls()
147
+ loaded_documents.extend(loader.load(uploaded_file))
148
 
149
  # all_loaders = [pdf_loader, markdown_loader, text_loader, csv_loader, python_loader, epub_loader, html_loader, ppt_loader, pptx_loader, doc_loader, docx_loader, odt_loader, notebook_loader]
150