Spaces:
Build error
Build error
Update app.py
Browse files
app.py
CHANGED
@@ -194,7 +194,7 @@ def _inference_ner_spancat(text, limit_outputs=10):
|
|
194 |
for doc in nlp.pipe(text):
|
195 |
spans = doc.spans["sc"]
|
196 |
#comp_raw_text = dict( sorted( dict(zip([str(x) for x in spans],[float(x)*penalty for x in spans.attrs['scores']])).items(), key=lambda x: x[1], reverse=True) )
|
197 |
-
company_list = list(set([str(span).replace('\'s', '').replace('
|
198 |
out.append(get_company_sectors(company_list))
|
199 |
return out
|
200 |
|
|
|
194 |
for doc in nlp.pipe(text):
|
195 |
spans = doc.spans["sc"]
|
196 |
#comp_raw_text = dict( sorted( dict(zip([str(x) for x in spans],[float(x)*penalty for x in spans.attrs['scores']])).items(), key=lambda x: x[1], reverse=True) )
|
197 |
+
company_list = list(set([str(span).replace('\'s', '').replace('\u2019s','') for span in filter_spans(spans, keep_longest=True)]))[:limit_outputs]
|
198 |
out.append(get_company_sectors(company_list))
|
199 |
return out
|
200 |
|