Update README.md
Browse files
README.md
CHANGED
@@ -104,9 +104,8 @@ from transformers import AutoModelForTokenClassification, AutoTokenizer, pipelin
|
|
104 |
MODEL_NAME = "impresso-project/ner-stacked-bert-multilingual"
|
105 |
|
106 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
107 |
-
model = AutoModelForTokenClassification.from_pretrained(MODEL_NAME, trust_remote_code=True)
|
108 |
|
109 |
-
ner_pipeline = pipeline("generic-ner", model=
|
110 |
|
111 |
sentence = "En l'an 1348, au plus fort des ravages de la peste noire à travers l'Europe, le Royaume de France se trouvait à la fois au bord du désespoir et face à une opportunité. À la cour du roi Philippe VI, les murs du Louvre étaient animés par les rapports sombres venus de Paris et des villes environnantes. La peste ne montrait aucun signe de répit, et le chancelier Guillaume de Nogaret, le conseiller le plus fidèle du roi, portait le lourd fardeau de gérer la survie du royaume."
|
112 |
entities = ner_pipeline(sentence)
|
|
|
104 |
MODEL_NAME = "impresso-project/ner-stacked-bert-multilingual"
|
105 |
|
106 |
tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
|
|
|
107 |
|
108 |
+
ner_pipeline = pipeline("generic-ner", model=MODEL_NAME, tokenizer=tokenizer, trust_remote_code=True, device='cpu')
|
109 |
|
110 |
sentence = "En l'an 1348, au plus fort des ravages de la peste noire à travers l'Europe, le Royaume de France se trouvait à la fois au bord du désespoir et face à une opportunité. À la cour du roi Philippe VI, les murs du Louvre étaient animés par les rapports sombres venus de Paris et des villes environnantes. La peste ne montrait aucun signe de répit, et le chancelier Guillaume de Nogaret, le conseiller le plus fidèle du roi, portait le lourd fardeau de gérer la survie du royaume."
|
111 |
entities = ner_pipeline(sentence)
|