Update app.py

app.py

@@ -2,7 +2,18 @@ import gradio as gr
 from transformers import MarianMTModel, MarianTokenizer
 
 def translate(text, target_language):
-    model_name = f'Helsinki-NLP/opus-mt-en-{target_language}'
+    language_codes = {
+        "Spanish": "es",
+        "French": "fr",
+        "Italian": "it",
+        "Ukrainian": "uk",
+        "Brazilian Portuguese": "pt",
+        "Japanese": "ja",
+        "Russian": "ru",
+        "Chinese": "zh"
+    }
+    target_language_code = language_codes[target_language]
+    model_name = f'helsinki-nlp/opus-mt-en-{target_language_code}'
     tokenizer = MarianTokenizer.from_pretrained(model_name)
     model = MarianMTModel.from_pretrained(model_name)
     inputs = tokenizer(text, return_tensors="pt")