Update app.py

app.py

@@ -4,15 +4,15 @@ from huggingface_hub import hf_hub_download
 from fastai.learner import load_learner
 from transformers import *
 
-pretrained_model_name = "Helsinki-NLP/opus-mt-en-ml"
-model_cls = AutoModelForSeq2SeqLM
-hf_arch, hf_config, hf_tokenizer, hf_model = BLURR.get_hf_objects(pretrained_model_name, model_cls=model_cls)
+def download():
+    pretrained_model_name = "Helsinki-NLP/opus-mt-en-ml"
+    model_cls = AutoModelForSeq2SeqLM
+    hf_arch, hf_config, hf_tokenizer, hf_model = BLURR.get_hf_objects(pretrained_model_name, model_cls=model_cls)
 
-model = load_learner(
-    hf_hub_download("kurianbenoy/kde_en_ml_translation_model", "saved_model.pkl")
-)
+    model = load_learner(hf_hub_download("kurianbenoy/kde_en_ml_translation_model", "saved_model.pkl"))
 
 def translate_text(text):
+    download()
     outputs = model.blurr_translate(text)
     return outputs[0]["translation_texts"]