pre-loading models

app.py

@@ -51,10 +51,17 @@ def load_model(source_language,target_language):
     return params
 
 #Load models of all available language pairs
+examples_available_models=[]
 model_mapping = {} 
 for availabe_lang in available_language_pairs:
-    model_mapping.update({availabe_lang:load_model(availabe_lang.split('-')[0],availabe_lang.split('-')[1])})
+    try:
+        model_mapping.update({availabe_lang:load_model(availabe_lang.split('-')[0],availabe_lang.split('-')[1])})
+        examples_available_models.append([f"{list(language_map.keys())[list(language_map.values()).index(availabe_lang.split('-')[0])]}",f"{list(language_map.keys())[list(language_map.values()).index(availabe_lang.split('-')[1])]}"]) #idea to extract key from value got from https://stackoverflow.com/questions/8023306/get-key-by-value-in-dictionary
+    except Exception:
+        continue
 
+if examples_available_models==[]:
+    raise Exception(f'Available models for Space cannot be empty!')
 
 
 def get_translation(source_language,target_language,source_sentence=None,source_file=None):
@@ -96,11 +103,7 @@ iface = gr.Interface(fn=get_translation,
   outputs=gr.outputs.Textbox(type="auto", label='Translation'),
   title=title,
   description=description,
-  examples=[
-        ['English','Swahili'],
-        ['English','Afrikaans'],['English','Arabic'],['Efik','English'],['English','Hausa'],
-        ['English','Igbo'],['English','Fon'],['English','Twi'],['Shona','English'],['Swahili','English'],
-        ['Yoruba','English']],
+  examples=examples_available_models,
   enable_queue=True,
   theme='huggingface')
 iface.launch()