1from transformers import AutoTokenizer, AutoModelForTokenClassification
2
3tokenizer = AutoTokenizer.from_pretrained("Jean-Baptiste/camembert-ner")
4model = AutoModelForTokenClassification.from_pretrained("Jean-Baptiste/camembert-ner")
5
6
7##### Process text sample (from wikipedia)
8
9from transformers import pipeline
10
11nlp = pipeline('ner', model=model, tokenizer=tokenizer, aggregation_strategy="simple")
12nlp("Apple est créée le 1er avril 1976 dans le garage de la maison d'enfance de Steve Jobs à Los Altos en Californie par Steve Jobs, Steve Wozniak et Ronald Wayne14, puis constituée sous forme de société le 3 janvier 1977 à l'origine sous le nom d'Apple Computer, mais pour ses 30 ans et pour refléter la diversification de ses produits, le mot « computer » est retiré le 9 janvier 2015.")
13
14
15[{'entity_group': 'ORG',
16 'score': 0.9472818374633789,
17 'word': 'Apple',
18 'start': 0,
19 'end': 5},
20 {'entity_group': 'PER',
21 'score': 0.9838564991950989,
22 'word': 'Steve Jobs',
23 'start': 74,
24 'end': 85},
25 {'entity_group': 'LOC',
26 'score': 0.9831605950991312,
27 'word': 'Los Altos',
28 'start': 87,
29 'end': 97},
30 {'entity_group': 'LOC',
31 'score': 0.9834540486335754,
32 'word': 'Californie',
33 'start': 100,
34 'end': 111},
35 {'entity_group': 'PER',
36 'score': 0.9841555754343668,
37 'word': 'Steve Jobs',
38 'start': 115,
39 'end': 126},
40 {'entity_group': 'PER',
41 'score': 0.9843501806259155,
42 'word': 'Steve Wozniak',
43 'start': 127,
44 'end': 141},
45 {'entity_group': 'PER',
46 'score': 0.9841533899307251,
47 'word': 'Ronald Wayne',
48 'start': 144,
49 'end': 157},
50 {'entity_group': 'ORG',
51 'score': 0.9468960364659628,
52 'word': 'Apple Computer',
53 'start': 243,
54 'end': 257}]
55