Update app.py

app.py

@@ -6,10 +6,21 @@ model_name = "Helsinki-NLP/opus-mt-en-ROMANCE"
 model = MarianMTModel.from_pretrained(model_name)
 tokenizer = MarianTokenizer.from_pretrained(model_name)
 
-def translate_text(text, target_lang):
+# Define language codes supported by the model
+language_codes = {
+    'French': 'fr',
+    'German': 'de',
+    'Italian': 'it',
+    'Portuguese': 'pt',
+    'Spanish': 'es',
+    # Add more languages if needed
+}
+
+def translate_text(text, target_lang_code):
     # Prepare the input and translate
     inputs = tokenizer.encode(text, return_tensors="pt")
-    translated = model.generate(inputs, decoder_start_token_id=tokenizer.lang_code_to_id[target_lang])
+    # Generate the translation
+    translated = model.generate(inputs, decoder_start_token_id=tokenizer.get_lang_id(target_lang_code))
     translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
     return translated_text
 
@@ -20,21 +31,12 @@ def main():
     text_to_translate = st.text_area("Enter text in English:")
 
     # Language options
-    languages = {
-        'French': 'fr',
-        'German': 'de',
-        'Italian': 'it',
-        'Portuguese': 'pt',
-        'Spanish': 'es',
-        # Add more languages if needed
-    }
-
-    selected_language = st.selectbox("Select target language:", list(languages.keys()))
+    selected_language = st.selectbox("Select target language:", list(language_codes.keys()))
 
     if st.button("Translate"):
         if text_to_translate:
-            target_lang = languages[selected_language]
-            translated_text = translate_text(text_to_translate, target_lang)
+            target_lang_code = language_codes[selected_language]
+            translated_text = translate_text(text_to_translate, target_lang_code)
             st.write(f"**Translation in {selected_language}:**")
             st.write(translated_text)
         else: