Fix runtime error

app.py

@@ -2,14 +2,9 @@ import os
 from dotenv import load_dotenv
 import gradio as gr
 from AinaTheme import theme
-from huggingface_hub import snapshot_download
-import nltk
 from api_endpoint import invoke_translate_endpoint
-
 from translate import translate_text
 
-nltk.download('punkt')
-
 load_dotenv()
 
 MODELS_PATH = "./models"
@@ -18,55 +13,36 @@ MAX_INPUT_CHARACTERS = int(os.environ.get("MAX_INPUT_CHARACTERS", default=1000))
 API_ENDPOINT_ENABLED = os.environ.get("API_ENDPOINT_ENABLED", default=True) == "True"
 
 
-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_ca_es = download_model("projecte-aina/aina-translator-ca-es", revision="main")
-model_dir_es_ca = download_model("projecte-aina/aina-translator-es-ca", revision="main")
-
-model_dir_ca_en = download_model("projecte-aina/aina-translator-ca-en", revision="main")
-model_dir_en_ca = download_model("projecte-aina/aina-translator-en-ca", revision="main")
-
-model_dir_ca_fr = download_model("projecte-aina/aina-translator-ca-fr", revision="main")
-model_dir_fr_ca = download_model("projecte-aina/aina-translator-fr-ca", revision="main")
-
-model_dir_ca_de = download_model("projecte-aina/aina-translator-ca-de", revision="main")
-
-model_dir_ca_it = download_model("projecte-aina/aina-translator-ca-it", revision="main")
-
-model_dir_ca_pt = download_model("projecte-aina/aina-translator-ca-pt", revision="main")
-model_dir_pt_ca = download_model("projecte-aina/aina-translator-pt-ca", revision="main")
-
 directions = {
     "Catalan": {
         "target": {
-            "Spanish": {"src": "ca", "tgt":"es", "model": (f"{os.path.join(MODELS_PATH, model_dir_ca_es)}/spm.model", f"{os.path.join(MODELS_PATH, model_dir_ca_es)}")},
-            "English": {"src": "ca", "tgt":"en", "model": (f"{os.path.join(MODELS_PATH, model_dir_ca_en)}/spm.model", f"{os.path.join(MODELS_PATH, model_dir_ca_en)}")},
-            "French": {"src": "ca", "tgt":"fr", "model": (f"{os.path.join(MODELS_PATH, model_dir_ca_fr)}/spm.model", f"{os.path.join(MODELS_PATH, model_dir_ca_fr)}")},
-            "German": {"src": "ca", "tgt":"de", "model": (f"{os.path.join(MODELS_PATH, model_dir_ca_de)}/spm.model", f"{os.path.join(MODELS_PATH, model_dir_ca_de)}")},
-            "Italian": {"src": "ca", "tgt":"it", "model": (f"{os.path.join(MODELS_PATH, model_dir_ca_it)}/spm.model", f"{os.path.join(MODELS_PATH, model_dir_ca_it)}")},
-            "Portuguese": {"src": "ca", "tgt":"pt", "model": (f"{os.path.join(MODELS_PATH, model_dir_ca_pt)}/spm.model", f"{os.path.join(MODELS_PATH, model_dir_ca_pt)}")}
+            "Spanish": {"src": "ca", "tgt":"es"},
+            "English": {"src": "ca", "tgt":"en"},
+            "French": {"src": "ca", "tgt":"fr"},
+            "German": {"src": "ca", "tgt":"de"},
+            "Italian": {"src": "ca", "tgt":"it"},
+            "Portuguese": {"src": "ca", "tgt":"pt"}
 
         }
     },
     "Spanish": {
         "target": {
-            "Catalan": {"src": "es", "tgt":"ca", "model": (f"{os.path.join(MODELS_PATH, model_dir_es_ca)}/spm.model", f"{os.path.join(MODELS_PATH, model_dir_es_ca)}")},
+            "Catalan": {"src": "es", "tgt":"ca"},
         }
     },
     "English": {
         "target": {
-            "Catalan": {"src": "en", "tgt":"ca", "model": (f"{os.path.join(MODELS_PATH, model_dir_en_ca)}/spm.model", f"{os.path.join(MODELS_PATH, model_dir_en_ca)}")},
+            "Catalan": {"src": "en", "tgt":"ca"},
         }
     },
     "French": {
         "target": {
-            "Catalan": {"src": "fr", "tgt":"ca", "model": (f"{os.path.join(MODELS_PATH, model_dir_fr_ca)}/spm.model", f"{os.path.join(MODELS_PATH, model_dir_fr_ca)}")},
+            "Catalan": {"src": "fr", "tgt":"ca"},
         }
     },
     "Portuguese": {
         "target": {
-            "Catalan": {"src": "pt", "tgt":"ca", "model": (f"{os.path.join(MODELS_PATH, model_dir_pt_ca)}/spm.model", f"{os.path.join(MODELS_PATH, model_dir_pt_ca)}")},
+            "Catalan": {"src": "pt", "tgt":"ca"},
         }
     }
 }
@@ -106,13 +82,11 @@ def translate_input(input, source_language, target_language):
     
     target_language_model = get_target_languege_model(source_language, target_language)
 
-    if (API_ENDPOINT_ENABLED and 'src' in target_language_model):
-         translation =  invoke_translate_endpoint(input, target_language_model)
-         if translation is not None:
-            return translation
-    translation = translate(input, source_language, target_language_model.get('model'))
-
-    return translation
+    translation =  invoke_translate_endpoint(input, target_language_model)
+    if translation is not None:
+        return translation
+    
+    return None
 
 
 def clear():