Update app.py

app.py

@@ -1,22 +1,18 @@
-from transformers import MarianMTModel, MarianTokenizer
-from huggingface_hub import login
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
 import gradio as gr
 
-# Add your Hugging Face token here
-login(token="huggingface_access_token")  # توکن خود را اینجا قرار دهید
+# بارگذاری مدل و توکنایزر
+model_name = "Persian-English-Translator/en2fa"
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
 
-# Load model and tokenizer
-model_name = "Helsinki-NLP/opus-mt-en-pes"  # یا هر مدل دیگری که استفاده می‌کنید
-tokenizer = MarianTokenizer.from_pretrained(model_name)
-model = MarianMTModel.from_pretrained(model_name)
-
-# Translation function
+# تابع ترجمه
 def translate(text):
-    inputs = tokenizer(text, return_tensors="pt", padding=True, truncation=True)
-    translated = model.generate(**inputs)
+    inputs = tokenizer.encode(text, return_tensors="pt", padding=True, truncation=True)
+    translated = model.generate(inputs)
     return tokenizer.decode(translated[0], skip_special_tokens=True)
 
-# Gradio Interface
+# رابط کاربری Gradio
 interface = gr.Interface(
     fn=translate,
     inputs=gr.Textbox(lines=5, placeholder="Enter English text here..."),