diff --git a/allegro.py b/allegro.py
index 53b9036..5f221f1 100644
--- a/allegro.py
+++ b/allegro.py
@@ -1,8 +1,7 @@
-from transformers import MarianForCausalLM, MarianTokenizer, TrainingArguments
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-model_name = "allegro/multislav-5lang"
-model = MarianForCausalLM.from_pretrained(model_name)
-tokenizer = MarianTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained("allegro/multislav-5lang")
+tokenizer = AutoTokenizer.from_pretrained("allegro/multislav-5lang")
 
 model.save_pretrained("./models/ably")
 tokenizer.save_pretrained("./models/ably")