AIEcosystem commited on
Commit
c00a43d
·
verified ·
1 Parent(s): b3a6a4d

Update src/streamlit_app.py

Browse files
Files changed (1) hide show
  1. src/streamlit_app.py +2 -2
src/streamlit_app.py CHANGED
@@ -421,8 +421,8 @@ def process_chunked_text(text, labels, model):
421
  chunks = chunk_text(text, max_chunk_size=MAX_CHUNK_CHARS)
422
  all_entities = []
423
 
424
- for chunk_text, chunk_offset in chunks:
425
- chunk_entities = model.predict_entities(chunk_text, labels)
426
  for entity in chunk_entities:
427
  entity['start'] += chunk_offset
428
  entity['end'] += chunk_offset
 
421
  chunks = chunk_text(text, max_chunk_size=MAX_CHUNK_CHARS)
422
  all_entities = []
423
 
424
+ for chunk_data, chunk_offset in chunks:
425
+ chunk_entities = model.predict_entities(chunk_data, labels)
426
  for entity in chunk_entities:
427
  entity['start'] += chunk_offset
428
  entity['end'] += chunk_offset