devve1 commited on
Commit
b256ccf
1 Parent(s): 7206d43

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -266,6 +266,7 @@ def load_models_and_documents():
266
  for doc in docs:
267
  texts.append(doc.page_content)
268
  del doc.metadata['title']
 
269
  metadatas.append(doc.metadata)
270
 
271
  docs_texts, docs_metadatas, dense_embeddings, sparse_embeddings = chunk_documents(texts, metadatas, dense_model, sparse_model)
@@ -395,7 +396,7 @@ if __name__ == '__main__':
395
  elements = partition(file=uploaded_file,
396
  strategy='hi_res',
397
  skip_infer_table_types=['png', 'pdf', 'jpg', 'xls', 'xlsx', 'heic'],
398
- hi_res_model_name='chipper',
399
  include_page_breaks=True
400
  )
401
 
 
266
  for doc in docs:
267
  texts.append(doc.page_content)
268
  del doc.metadata['title']
269
+ del doc.metadata['summary']
270
  metadatas.append(doc.metadata)
271
 
272
  docs_texts, docs_metadatas, dense_embeddings, sparse_embeddings = chunk_documents(texts, metadatas, dense_model, sparse_model)
 
396
  elements = partition(file=uploaded_file,
397
  strategy='hi_res',
398
  skip_infer_table_types=['png', 'pdf', 'jpg', 'xls', 'xlsx', 'heic'],
399
+ hi_res_model_name='yolox',
400
  include_page_breaks=True
401
  )
402