Spaces:
Runtime error
Runtime error
Update src/streamlit_app.py
Browse files- 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
|
| 425 |
-
chunk_entities = model.predict_entities(
|
| 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
|