Create app.py

app.py

@@ -0,0 +1,26 @@
+from transformers import MBartForConditionalGeneration, MBart50Tokenizer
+import gradio as gr
+
+# Load Facebook's mbart-large-50 model and tokenizer
+model_name = "facebook/mbart-large-50-many-to-one-mmt"
+tokenizer = MBart50Tokenizer.from_pretrained(model_name)
+model = MBartForConditionalGeneration.from_pretrained(model_name)
+
+def translate_tamil_to_english(tamil_text):
+  """Translates Tamil text to English using the mbart-large-50 model."""
+
+  tokenizer.src_lang = "ta_IN"
+  inputs = tokenizer(tamil_text, return_tensors="pt")
+  translated_tokens = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id["en_XX"])
+  translated_text = tokenizer.batch_decode(translated_tokens, skip_special_tokens=True)[0]
+  return translated_text
+
+iface = gr.Interface(
+    fn=translate_tamil_to_english,
+    inputs=gr.Textbox(lines=2, placeholder="Enter Tamil text here..."),
+    outputs="text",
+    title="Tamil to English Translator",
+    description="Translate Tamil text to English using Facebook's mbart-large-50 model.",
+)
+
+iface.launch() # Remove share=True when deploying to Spaces