socialcomp commited on
Commit
61b6f64
1 Parent(s): 4c64fb0

Update predict.py

Browse files
Files changed (1) hide show
  1. 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/RealMedNLP_CR_JA", [sentences_norm], len(id_to_tags))
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]):