diff --git a/nominal_sense_srl/predictor.py b/nominal_sense_srl/predictor.py index 0cf42d9..1d1907f 100644 --- a/nominal_sense_srl/predictor.py +++ b/nominal_sense_srl/predictor.py @@ -13,7 +13,7 @@ from nominal_srl.nominal_srl_reader import separate_hyphens -@Predictor.register("nombank-sense-srl") +@Predictor.register("nominal-sense-srl") class NomSenseSRLPredictor(Predictor): """ Predictor for the nominal BERT-based SRL model.