Update app.py

app.py

@@ -1,16 +1,17 @@
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
+from transformers import MT5ForConditionalGeneration, MT5Tokenizer
 import gradio as gr
 
 # بارگذاری مدل و توکنایزر
-model_name = "Persian-English-Translator/en2fa"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+model_name = "persiannlp/mt5-small-parsinlu-translation_en_fa"
+tokenizer = MT5Tokenizer.from_pretrained(model_name)
+model = MT5ForConditionalGeneration.from_pretrained(model_name)
 
 # تابع ترجمه
 def translate(text):
-    inputs = tokenizer.encode(text, return_tensors="pt", padding=True, truncation=True)
-    translated = model.generate(inputs)
-    return tokenizer.decode(translated[0], skip_special_tokens=True)
+    input_ids = tokenizer.encode(text, return_tensors="pt")
+    outputs = model.generate(input_ids)
+    translated_text = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return translated_text
 
 # رابط کاربری Gradio
 interface = gr.Interface(