Update app.py

app.py

@@ -4,10 +4,17 @@ import subprocess
 
 from transformers import AutoModel
 
-# NOS-MT-en-gl.p
+MODELS_PATH = "./models"
+HF_CACHE_DIR = "./hf_cache"
+
+def download_model(repo_id, revision="main"):
+    return snapshot_download(repo_id=repo_id, revision=revision, local_dir=os.path.join(MODELS_PATH, repo_id), cache_dir=HF_CACHE_DIR)
+
+model_dir_gl_en = download_model("proxectonos/Nos_MT-OpenNMT-gl-en", revision="main")
+model_dir_en_gl = download_model("proxectonos/Nos_MT-OpenNMT-en-gl", revision="main")
 
 def translate(input_text):
-    command = f"onmt_translate -src {input_text} -model NOS-MT-en-gl.pt --output ./output_file.txt --replace_unk"
+    command = f"onmt_translate -src {input_text} -model model_dir_en_gl --output ./output_file.txt --replace_unk"
     process = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
     stdout, stderr = process.communicate()
     if process.returncode != 0: