captain-awesome commited on
Commit
8cfaaa3
·
1 Parent(s): 880820c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -4
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 = "." + file_path.rsplit(".", 1)[-1]
124
  if ext in mapping:
125
  loader_class, loader_args = mapping[ext]
126
- loader = loader_class(file_path, **loader_args)
127
  else:
128
- loader = default_loader(file_path)
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