captain-awesome commited on
Commit
85a9639
1 Parent(s): 25f639b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +17 -16
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
- FILE_LOADER_MAPPING = {
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