update
Browse files
app.py
CHANGED
@@ -3,7 +3,7 @@ from transformers import pipeline
|
|
3 |
from transformers import AutoTokenizer, AutoModelForTokenClassification
|
4 |
tokenizer = AutoTokenizer.from_pretrained("51la5/roberta-large-NER")
|
5 |
model = AutoModelForTokenClassification.from_pretrained("51la5/roberta-large-NER")
|
6 |
-
classifier = pipeline("ner", model=model, tokenizer=tokenizer
|
7 |
|
8 |
def get_ner(text):
|
9 |
output = classifier(text)
|
|
|
3 |
from transformers import AutoTokenizer, AutoModelForTokenClassification
|
4 |
tokenizer = AutoTokenizer.from_pretrained("51la5/roberta-large-NER")
|
5 |
model = AutoModelForTokenClassification.from_pretrained("51la5/roberta-large-NER")
|
6 |
+
classifier = pipeline("ner", model=model, tokenizer=tokenizer)
|
7 |
|
8 |
def get_ner(text):
|
9 |
output = classifier(text)
|