Update app.py

app.py

@@ -1,10 +1,10 @@
 import gradio as gr
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 
-# Load the model and tokenizer
-model_name = "Helsinki-NLP/opus-mt-en-th"
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-tokenizer = AutoTokenizer.from_pretrained(model_name)
+# Load the mBART model and tokenizer for multilingual translation
+model_name = "facebook/mbart-large-50-many-to-many-mmt"
+model = MBartForConditionalGeneration.from_pretrained(model_name)
+tokenizer = MBart50TokenizerFast.from_pretrained(model_name, src_lang="en_XX", tgt_lang="th_TH")
 
 # Prediction function
 def translate_text(input_text):