Views
No views yet
1import torch
2from transformers import AutoModelForTokenClassification, AutoTokenizer, AutoConfig, pipeline
3
4config_ner = AutoConfig.from_pretrained("erythropygia/bert-turkish-entityrecognition")
5model_ner = AutoModelForTokenClassification.from_pretrained("erythropygia/bert-turkish-entityrecognition",config=config_ner)
6tokenizer_ner= AutoTokenizer.from_pretrained("erythropygia/bert-turkish-entityrecognition")
7
8device = torch.device('cpu') #or 'cuda'
9
10ner_pipeline = pipeline(
11 task='ner',
12 model=model_ner,
13 tokenizer=tokenizer_ner,
14 framework='pt',
15 #device=device
16)
17
18input = "Bugün Fenerbahçe maçına gideceğim. Maçtan sonra Mehmet Yıldırım'la Bostancıda buluşacağım"
19
20ner_result = ner_pipeline(input)
21print(ner_result)