| from transformers import CamembertTokenizer, CamembertForTokenClassification, TokenClassificationPipeline | |
| OUTPUT_PATH = './' | |
| tokenizer = CamembertTokenizer.from_pretrained(OUTPUT_PATH) | |
| model = CamembertForTokenClassification.from_pretrained(OUTPUT_PATH) | |
| pos = TokenClassificationPipeline(model=model, tokenizer=tokenizer) | |
| def make_prediction(sentence): | |
| labels = [l['entity'] for l in pos(sentence)] | |
| return list(zip(sentence.split(" "), labels)) | |
| res = make_prediction("George Washington est allé à Washington") | |