Updated model and tokenizer HugingFace loading with Auto... clases

modules/module_languageModel.py

@@ -1,5 +1,5 @@
 # --- Imports libs ---
-from transformers import BertForMaskedLM, BertTokenizer
+from transformers import AutoTokenizer, AutoModelForMaskedLM
 
 class LanguageModel:
     def __init__(
@@ -7,9 +7,9 @@ class LanguageModel:
         model_name: str
     ) -> None:
      
-        print("Download language model...")
-        self.__tokenizer = BertTokenizer.from_pretrained(model_name)
-        self.__model = BertForMaskedLM.from_pretrained(model_name, return_dict=True)
+        print("Downloading language model...")
+        self.__tokenizer = AutoTokenizer.from_pretrained(model_name)
+        self.__model = AutoModelForMaskedLM.from_pretrained(model_name)
     
     def initTokenizer(
         self