Spaces:
Runtime error
Runtime error
Commit
·
9097403
1
Parent(s):
bf1769e
Update app.py
Browse files
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 |
-
|
144 |
-
|
145 |
-
|
146 |
-
|
147 |
-
|
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 |
|