a | b/HealthCare_ChatBot/script.py | ||
---|---|---|---|
1 | from transformers import pipeline |
||
2 | PRETRAINED = "raynardj/ner-disease-ncbi-bionlp-bc5cdr-pubmed" |
||
3 | ners = pipeline(task="ner",model=PRETRAINED, tokenizer=PRETRAINED) |
||
4 | |||
5 | import pickle as pk |
||
6 | with open('newsave_model', 'wb') as f: |
||
7 | pk.dump(ners, f) |