Views
No views yet
1from transformers import AutoTokenizer, AutoModelForTokenClassification
2
3model_id = "zexho/uae_address_roberta_v1.0.3"
4
5tokenizer = AutoTokenizer.from_pretrained(model_id)
6model = AutoModelForTokenClassification.from_pretrained(model_id, ignore_mismatched_sizes=True)
7
8text = "مثال على العنوان"
9inputs = tokenizer(text.split(), is_split_into_words=True, return_tensors="pt", truncation=True)
10outputs = model(**inputs)BertForTokenClassification 对齐;若出现不匹配,可设置 ignore_mismatched_sizes=True 或自行适配。labels.json 包含 id2label 与 label2id 用于解释预测结果。