socialcomp
commited on
Commit
•
61b6f64
1
Parent(s):
4c64fb0
Update predict.py
Browse files- predict.py +1 -1
predict.py
CHANGED
@@ -131,7 +131,7 @@ if __name__ == '__main__':
|
|
131 |
sentences_raw = [s for s in re.split(r'\n', articles_raw) if s != '']
|
132 |
sentences_norm = [s for s in re.split(r'\n', article_norm) if s != '']
|
133 |
|
134 |
-
text_entities_set = predict_entities("sociocom/
|
135 |
|
136 |
|
137 |
for i, texts_ent in enumerate(text_entities_set[0]):
|
|
|
131 |
sentences_raw = [s for s in re.split(r'\n', articles_raw) if s != '']
|
132 |
sentences_norm = [s for s in re.split(r'\n', article_norm) if s != '']
|
133 |
|
134 |
+
text_entities_set = predict_entities("sociocom/MedNER-CR-JA", [sentences_norm], len(id_to_tags))
|
135 |
|
136 |
|
137 |
for i, texts_ent in enumerate(text_entities_set[0]):
|