pmaitra commited on
Commit
f90f6b0
1 Parent(s): 19b367a

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +6 -5
README.md CHANGED
@@ -64,14 +64,15 @@ Developing a BioBERT-based Natural Language Processing Algorithm for Acute Myelo
64
  <div class="wrapper">
65
  <span class="inner">
66
 
67
- from transformers import pipeline, AutoTokenizer, AutoModelForTokenClassification
68
 
 
 
69
 
70
- tokenizer = AutoTokenizer.from_pretrained("pmaitra/en_biobert_ner_symptom")
71
- model = AutoModelForTokenClassification.from_pretrained("pmaitra/en_biobert_ner_symptom")
72
 
73
- pipe = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
74
- print(pipe("He complained of dizziness during the trip."))
75
 
76
  </span>
77
  </div>
 
64
  <div class="wrapper">
65
  <span class="inner">
66
 
67
+ !pip install https://huggingface.co/pmaitra/en_biobert_ner_symptom/resolve/main/en_biobert_ner_symptom-any-py3-none-any.whl
68
 
69
+ import spacy
70
+ nlp = spacy.load("en_biobert_ner_symptom")
71
 
72
+ doc = nlp("He complained of dizziness and nausea during the Iowa trip.")
 
73
 
74
+ for ent in doc.ents:
75
+ print(ent)
76
 
77
  </span>
78
  </div>