Update app.py

app.py

@@ -2,7 +2,7 @@ import streamlit as st
 from transformers import MarianMTModel, MarianTokenizer
 
 # Load the MarianMT model and tokenizer
-model_name = "Helsinki-NLP/opus-mt-en-ROMANCE"
+model_name = "Helsinki-NLP/opus-mt-en-many-to-many-mmt"
 model = MarianMTModel.from_pretrained(model_name)
 tokenizer = MarianTokenizer.from_pretrained(model_name)
 
@@ -13,6 +13,14 @@ language_codes = {
     'Italian': 'it',
     'Portuguese': 'pt',
     'Spanish': 'es',
+    'Chinese': 'zh',
+    'Japanese': 'ja',
+    'Korean': 'ko',
+    'Russian': 'ru',
+    'Arabic': 'ar',
+    'Turkish': 'tr',
+    'Hindi': 'hi',
+    'Swahili': 'sw',
     # Add more languages if needed
 }
 
@@ -20,12 +28,12 @@ def translate_text(text, target_lang_code):
     # Prepare the input and translate
     inputs = tokenizer.encode(text, return_tensors="pt")
     # Generate the translation
-    translated = model.generate(inputs, decoder_start_token_id=tokenizer.get_lang_id(target_lang_code))
+    translated = model.generate(inputs, decoder_start_token_id=tokenizer.lang_code_to_id[target_lang_code])
     translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
     return translated_text
 
 def main():
-    st.title("English to Any Language Translator")
+    st.title("Multilingual Translator")
 
     # User input for English text
     text_to_translate = st.text_area("Enter text in English:")