update based on revised joeynmt

app.py

@@ -1,6 +1,6 @@
 import gradio as gr
 import yaml
-from joeynmt.prediction import load_params_for_prediction,translate
+from joeynmt.prediction import load_params_for_prediction,translate_for_hf_space
 from huggingface_hub import hf_hub_download
 
 language_map = {'English':'en','Swahili':'sw','Fon':'fon','Igbo':'ig',
@@ -68,7 +68,6 @@ if examples_available_models==[]:
 def get_translation(source_language,target_language,source_sentence=None,source_file=None):
     '''
     This takes a sentence and gets the translation.
-    type_=2 tells joeynmt translate that it should expect a sentence. 
     '''
     
     source_language_ = language_map[source_language]
@@ -76,13 +75,13 @@ def get_translation(source_language,target_language,source_sentence=None,source_
    
     
     source = source_sentence
-    type_=2
+    translation_type='sentence'
     if source_file!=None:
-        type_=1
+        translation_type='file'
         source = source_file.name
     try:     
         params = model_mapping[f'{source_language_}-{target_language_}']
-        pred = translate(params,source,type_)
+        pred = translate_for_hf_space(params,source,translation_type)
     except Exception:
         return f'There was an issue loading the translation model for {source_language} -> {target_language}. Try another pair please'