Update app.py

app.py

@@ -6,18 +6,18 @@ text_translator = pipeline("translation", model="facebook/nllb-200-distilled-600
 
 # Dictionary mapping destination languages to their FLORES-200 codes
 language_mapping = {
-    "German": "German",
-    "Eastern Panjabi": "Punjabi",
-    "Sanskrit": "Sanskrit",
-    "Urdu": "Urdu",
-    "Tamil": "Tamil",
-    "Telugu": "Telugu",
-    "Yue Chinese": "Chinese",
-    "Chinese (Simplified)": "Chinese",
-    "Chinese (Traditional)": "Chinese",
-    "Hindi": "Hindi",
-    "French": "French",
-    "Spanish": "Spanish"
+    "German": "de_DE",
+    "Eastern Panjabi": "pa_IN",
+    "Sanskrit": "sa_IN",
+    "Urdu": "ur",
+    "Tamil": "ta",
+    "Telugu": "te",
+    "Yue Chinese": "zh_yue",
+    "Chinese (Simplified)": "zh",
+    "Chinese (Traditional)": "zh",
+    "Hindi": "hi",
+    "French": "fr",
+    "Spanish": "es"
 }
 
 def translate_text(text, destination_language):
@@ -37,11 +37,10 @@ def translate_text(text, destination_language):
         return "Unsupported language selected. Please choose a valid language."
 
     try:
-        # Format translation prompt for T5 model
-        translation_prompt = f"translate English to {dest_code}: {text}"
-        
         # Perform translation using T5 model pipeline
-        translation = text_translator(translation_prompt)
+        translation = text_translator(text,
+                                      src_lang="en",
+                                      tgt_lang=dest_code)
         translated_text = translation[0]["translation_text"]
         
         # Optionally return confidence score