Update app.py

app.py

@@ -1,17 +1,16 @@
 # Import necessary modules
 import gradio as gr
-from transformers import MBartForConditionalGeneration, MBart50Tokenizer
+from transformers import MarianMTModel, MarianTokenizer
 
 # Load the English to Urdu translation model
-model_name = "abdulwaheed1/english-to-urdu-translation-mbart"
-tokenizer = MBart50Tokenizer.from_pretrained(model_name)
-model = MBartForConditionalGeneration.from_pretrained(model_name)
+model_name = "Helsinki-NLP/opus-mt-en-ur"
+tokenizer = MarianTokenizer.from_pretrained(model_name)
+model = MarianMTModel.from_pretrained(model_name)
 
 # Define the translation function
 def translate(text):
-    tokenizer.src_lang = "en_XX"  # Set source language as English
     inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
-    translated = model.generate(**inputs, forced_bos_token_id=tokenizer.lang_code_to_id["ur_PK"])
+    translated = model.generate(**inputs)
     output = tokenizer.decode(translated[0], skip_special_tokens=True)
     return output