[a5cbc8]: / HealthCare_ChatBot / script.py

Download this file

8 lines (6 with data), 251 Bytes

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