Spaces:

bacngv
/

roberta-ner

Runtime error

push

9981175 verified over 2 years ago

281 Bytes

from xlmr_for_token_classification import XLMRForTokenClassification
from collections import namedtuple

Config = namedtuple('config', ['pretrained_path', 'n_labels', 'hidden_size'])
config = Config('pretrained_models/xlmr.base', 5, 768)

model = XLMRForTokenClassification(config)

	from xlmr_for_token_classification import XLMRForTokenClassification
	from collections import namedtuple

	Config = namedtuple('config', ['pretrained_path', 'n_labels', 'hidden_size'])
	config = Config('pretrained_models/xlmr.base', 5, 768)

	model = XLMRForTokenClassification(config)