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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -121,7 +121,8 @@ def load_document(
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)
 
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
+ ext = os.path.splitext(uploaded_file.name)[-1][1:].lower()
126
  if ext in mapping:
127
  loader_class, loader_args = mapping[ext]
128
  loader = loader_class(uploaded_files, **loader_args)