Views
No views yet
1from transformers import AutoTokenizer,AutoModelForTokenClassification,TokenClassificationPipeline
2tokenizer=AutoTokenizer.from_pretrained("KoichiYasuoka/bert-base-vietnamese-upos")
3model=AutoModelForTokenClassification.from_pretrained("KoichiYasuoka/bert-base-vietnamese-upos")
4pipeline=TokenClassificationPipeline(tokenizer=tokenizer,model=model,aggregation_strategy="simple")
5nlp=lambda x:[(x[t["start"]:t["end"]],t["entity_group"]) for t in pipeline(x)]
6print(nlp("Hai cái đầu thì tốt hơn một."))1import esupar
2nlp=esupar.load("KoichiYasuoka/bert-base-vietnamese-upos")
3print(nlp("Hai cái đầu thì tốt hơn một."))