Update app.py

app.py

@@ -1,13 +1,13 @@
 import streamlit as st
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
 
 def main():
     st.title("Translation App")
 
     # Load model and tokenizer
-    model_name = "facebook/mbart-large-50-one-to-many-mmt"
-    model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
-    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-one-to-many-mmt")
+    tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-one-to-many-mmt", src_lang="en_XX")
 
     # Input text area
     input_text = st.text_area("Enter text to translate", "")
@@ -22,9 +22,9 @@ def main():
 
 def translate_text(input_text, model, tokenizer):
     # Tokenize input text
-    model_inputs = tokenizer(input_text, return_tensors="pt").input_ids
+    input_ids = tokenizer(input_text, return_tensors="pt").input_ids
     generated_tokens = model.generate(
-    **model_inputs,
+    **input_ids,
     forced_bos_token_id=tokenizer.lang_code_to_id["hi_IN"]
 )