captain-awesome
commited on
Commit
•
85a9639
1
Parent(s):
25f639b
Update app.py
Browse files
app.py
CHANGED
@@ -73,22 +73,7 @@ from langchain.embeddings import HuggingFaceBgeEmbeddings
|
|
73 |
from io import BytesIO
|
74 |
from langchain.document_loaders import PyPDFLoader
|
75 |
|
76 |
-
|
77 |
-
"csv": (CSVLoader, {"encoding": "utf-8"}),
|
78 |
-
"doc": (UnstructuredWordDocumentLoader, {}),
|
79 |
-
"docx": (UnstructuredWordDocumentLoader, {}),
|
80 |
-
"epub": (UnstructuredEPubLoader, {}),
|
81 |
-
"html": (UnstructuredHTMLLoader, {}),
|
82 |
-
"md": (UnstructuredMarkdownLoader, {}),
|
83 |
-
"odt": (UnstructuredODTLoader, {}),
|
84 |
-
"pdf": (PyPDFLoader, {}),
|
85 |
-
"ppt": (UnstructuredPowerPointLoader, {}),
|
86 |
-
"pptx": (UnstructuredPowerPointLoader, {}),
|
87 |
-
"txt": (TextLoader, {"encoding": "utf8"}),
|
88 |
-
"ipynb": (NotebookLoader, {}),
|
89 |
-
"py": (PythonLoader, {}),
|
90 |
-
# Add more mappings for other file extensions and loaders as needed
|
91 |
-
}
|
92 |
|
93 |
|
94 |
# def load_model():
|
@@ -291,6 +276,22 @@ FILE_LOADER_MAPPING = {
|
|
291 |
# return result, sources
|
292 |
|
293 |
def main():
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
294 |
|
295 |
|
296 |
|
|
|
73 |
from io import BytesIO
|
74 |
from langchain.document_loaders import PyPDFLoader
|
75 |
|
76 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
77 |
|
78 |
|
79 |
# def load_model():
|
|
|
276 |
# return result, sources
|
277 |
|
278 |
def main():
|
279 |
+
FILE_LOADER_MAPPING = {
|
280 |
+
"csv": (CSVLoader, {"encoding": "utf-8"}),
|
281 |
+
"doc": (UnstructuredWordDocumentLoader, {}),
|
282 |
+
"docx": (UnstructuredWordDocumentLoader, {}),
|
283 |
+
"epub": (UnstructuredEPubLoader, {}),
|
284 |
+
"html": (UnstructuredHTMLLoader, {}),
|
285 |
+
"md": (UnstructuredMarkdownLoader, {}),
|
286 |
+
"odt": (UnstructuredODTLoader, {}),
|
287 |
+
"pdf": (PyPDFLoader, {}),
|
288 |
+
"ppt": (UnstructuredPowerPointLoader, {}),
|
289 |
+
"pptx": (UnstructuredPowerPointLoader, {}),
|
290 |
+
"txt": (TextLoader, {"encoding": "utf8"}),
|
291 |
+
"ipynb": (NotebookLoader, {}),
|
292 |
+
"py": (PythonLoader, {}),
|
293 |
+
# Add more mappings for other file extensions and loaders as needed
|
294 |
+
}
|
295 |
|
296 |
|
297 |
|