Spaces:
Runtime error
Runtime error
Commit
·
8cfaaa3
1
Parent(s):
880820c
Update app.py
Browse files
app.py
CHANGED
@@ -115,17 +115,18 @@ def load_model():
|
|
115 |
return llm
|
116 |
|
117 |
def load_document(
|
118 |
-
file_path: str,
|
|
|
119 |
mapping: dict = FILE_LOADER_MAPPING,
|
120 |
default_loader: BaseLoader = UnstructuredFileLoader,
|
121 |
) -> Document:
|
122 |
# Choose loader from mapping, load default if no match found
|
123 |
-
ext = "." +
|
124 |
if ext in mapping:
|
125 |
loader_class, loader_args = mapping[ext]
|
126 |
-
loader = loader_class(
|
127 |
else:
|
128 |
-
loader = default_loader(
|
129 |
loaded_documents = []
|
130 |
loaded_documents.extend(loader.load())
|
131 |
return loaded_documents
|
|
|
115 |
return llm
|
116 |
|
117 |
def load_document(
|
118 |
+
# file_path: str,
|
119 |
+
uploaded_files: str,
|
120 |
mapping: dict = FILE_LOADER_MAPPING,
|
121 |
default_loader: BaseLoader = UnstructuredFileLoader,
|
122 |
) -> Document:
|
123 |
# Choose loader from mapping, load default if no match found
|
124 |
+
ext = "." + uploaded_files.rsplit(".", 1)[-1]
|
125 |
if ext in mapping:
|
126 |
loader_class, loader_args = mapping[ext]
|
127 |
+
loader = loader_class(uploaded_files, **loader_args)
|
128 |
else:
|
129 |
+
loader = default_loader(uploaded_files)
|
130 |
loaded_documents = []
|
131 |
loaded_documents.extend(loader.load())
|
132 |
return loaded_documents
|