Update README.md

README.md

@@ -66,11 +66,12 @@ Developing a BioBERT-based Natural Language Processing Algorithm for Acute Myelo
 
 from transformers import pipeline, AutoTokenizer, AutoModelForTokenClassification
 
-tokenizer = AutoTokenizer.from_pretrained("d4data/biomedical-ner-all")
-model = AutoModelForTokenClassification.from_pretrained("d4data/biomedical-ner-all")
 
-pipe = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple") # pass device=0 if using gpu
-pipe("""The patient reported no recurrence of palpitations at follow-up 6 months after the ablation.""")
+tokenizer = AutoTokenizer.from_pretrained("pmaitra/en_biobert_ner_symptom")
+model = AutoModelForTokenClassification.from_pretrained("pmaitra/en_biobert_ner_symptom")
+
+pipe = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple") 
+print(pipe("He complained of dizziness during the trip."))
 
   </span>
 </div>