pre-loading models

app.py

@@ -11,7 +11,7 @@ language_map = {'English':'en','Swahili':'sw','Fon':'fon','Igbo':'ig',
                 
 #List of available languages I worked on.
 available_language_pairs =['en-sw','en-af','en-ar','efi-en','en-ha','en-ig','en-fon','en-twi','sn-en','sw-en','yo-en']
-
+available_languages = list(language_map.keys())
 
 def load_config(path="configs/default.yaml") -> dict:
     """
@@ -50,9 +50,10 @@ def load_model(source_language,target_language):
     params = load_params_for_prediction(parsed_yaml_file,best_ckpt)
     return params
 
+
 #Load models of all available language pairs
-examples_available_models=[]
 model_mapping = {} 
+examples_available_models=[] # Keep track of models that loaded successfully and display only them in the Examples.
 for availabe_lang in available_language_pairs:
     try:
         model_mapping.update({availabe_lang:load_model(availabe_lang.split('-')[0],availabe_lang.split('-')[1])})