Update app.py

app.py

@@ -72,7 +72,7 @@ def mtom(model_name, sl, tl, input_text):
     tokenizer.src_lang = sl
     encoded = tokenizer(input_text, return_tensors="pt")
     generated_tokens = model.generate(**encoded, forced_bos_token_id=tokenizer.get_lang_id(tl))
-    return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)
+    return tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
 
 def HelsinkiNLPAutoTokenizer(sl, tl, input_text):
     if model_name == "Helsinki-NLP":
@@ -286,7 +286,7 @@ def translate_text(input_text: str, sselected_language: str, tselected_language:
         elif model_name == 'Google':
             translated_text = google(sl, tl, input_text)
     
-        elif "m2m" in model_name:
+        elif "m2m" in model_name.lower():
             translated_text = mtom(model_name, sl, tl, input_text)
         
         elif model_name == "utter-project/EuroLLM-1.7B-Instruct":
@@ -330,7 +330,7 @@ def translate_text(input_text: str, sselected_language: str, tselected_language:
         translated_text = error
     return translated_text, message_text
 
-# Define a function to swap dropdown values
+# Function to swap dropdown values
 def swap_languages(src_lang, tgt_lang):
     return tgt_lang, src_lang