bugfix

predict.py

@@ -46,7 +46,7 @@ class Predictor:
         )
     
         self.model = DiacritizerD2(config)
-        self.model.sentence_diac.build(word_embeddings, vocab_size)
+        self.model.build(word_embeddings, vocab_size)
         state_dict = T.load(config["paths"]["load"], map_location=T.device(self.device))['state_dict']
         self.model.load_state_dict(state_dict)
         self.model.to(self.device)