Update app.py
Browse files
app.py
CHANGED
@@ -149,8 +149,14 @@ def ner_inference(txt):
|
|
149 |
def ner_inference_long_text(txt):
|
150 |
entities = []
|
151 |
doc = nlp(txt)
|
|
|
|
|
|
|
152 |
for sent in doc.sents:
|
153 |
entities.extend(ner_inference(sent.text))
|
|
|
|
|
|
|
154 |
return entities
|
155 |
|
156 |
|
|
|
149 |
def ner_inference_long_text(txt):
|
150 |
entities = []
|
151 |
doc = nlp(txt)
|
152 |
+
n_sents = len([_ for _ in doc.sents])
|
153 |
+
n = 0
|
154 |
+
progress_bar = st.progress(0, text=f'Processed 0 / {n_sents} sentences')
|
155 |
for sent in doc.sents:
|
156 |
entities.extend(ner_inference(sent.text))
|
157 |
+
n += 1
|
158 |
+
progress_bar.progress(n / n_sents, text=f'Processed {n} / {n_sents} sentences')
|
159 |
+
progress_bar.empty()
|
160 |
return entities
|
161 |
|
162 |
|