Update app.py

app.py

@@ -65,19 +65,19 @@ def HelsinkiNLPAutoTokenizer(sl, tl, input_text):
 def HelsinkiNLP(sl, tl, input_text):
     try:
         model_name = f"Helsinki-NLP/opus-mt-{sl}-{tl}"
-        pipe = pipeline("translation", model=model_name)
+        pipe = pipeline("translation", model=model_name, device=-1)
         # translation = pipe(input_text)
         # return translation[0]['translation_text'], f'Translated from {sl} to {tl} with {model_name}.'
     except EnvironmentError:
         try:   
             model_name = f"Helsinki-NLP/opus-tatoeba-{sl}-{tl}"
-            pipe = pipeline("translation", model=model_name)
+            pipe = pipeline("translation", model=model_name, device=-1)
             translation = pipe(input_text)
             return translation[0]['translation_text'], f'Translated from {sl} to {tl} with {model_name}.'
         except EnvironmentError as error:
             return f"Error finding model: {model_name}! Try other available language combination.", error
     except KeyError as error:
-        return f"Error: Translation direction {source_readable} to {target} is not supported by Helsinki Translation Models", error     
+        return f"Error: Translation direction {sl} to {tl} is not supported by Helsinki Translation Models", error     
 
 def flan(model_name, sl, tl, input_text):
     tokenizer = T5Tokenizer.from_pretrained(model_name, legacy=False)
@@ -214,7 +214,8 @@ def translate_text(input_text: str, sselected_language: str, tselected_language:
     print(message_text)
     
     if model_name.startswith("Helsinki-NLP"):
-        return translated_text, message_text = HelsinkiNLP(sl, tl, input_text)
+        translated_text, message_text = HelsinkiNLP(sl, tl, input_text)
+        return translated_text, message_text
         
     elif model_name == "utter-project/EuroLLM-1.7B-Instruct":
         translated_text = eurollm_instruct(model_name, sselected_language, tselected_language, input_text)