Create models.py

models.py

@@ -0,0 +1,19 @@
+from transformers import MarianMTModel, MarianTokenizer
+
+cache = {}
+
+def load_translation_model(model_name):
+if model_name in cache:
+return cache[model_name]
+
+tokenizer = MarianTokenizer.from_pretrained(model_name)
+model = MarianMTModel.from_pretrained(model_name)
+
+cache[model_name] = (model, tokenizer)
+return model, tokenizer
+
+def translate_text(text, model, tokenizer):
+batch = tokenizer([text], return_tensors="pt", padding=True)
+gen = model.generate(**batch)
+result = tokenizer.batch_decode(gen, skip_special_tokens=True)[0]
+return result