--- a +++ b/HealthCare_ChatBot/script.py @@ -0,0 +1,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)