awacke1 commited on
Commit
e43bb54
1 Parent(s): ed1a461

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ from collections import defaultdict
5
 
6
  # Create tokenizer for biomed model
7
  from transformers import pipeline, AutoTokenizer, AutoModelForTokenClassification
8
- tokenizer = AutoTokenizer.from_pretrained("d4data/biomedical-ner-all")
9
  model = AutoModelForTokenClassification.from_pretrained("d4data/biomedical-ner-all")
10
  pipe = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
11
 
 
5
 
6
  # Create tokenizer for biomed model
7
  from transformers import pipeline, AutoTokenizer, AutoModelForTokenClassification
8
+ tokenizer = AutoTokenizer.from_pretrained("d4data/biomedical-ner-all") # https://huggingface.co/d4data/biomedical-ner-all?text=asthma
9
  model = AutoModelForTokenClassification.from_pretrained("d4data/biomedical-ner-all")
10
  pipe = pipeline("ner", model=model, tokenizer=tokenizer, aggregation_strategy="simple")
11