Update app.py

app.py

@@ -5,11 +5,13 @@ import torch
 import gradio as gr
 import numpy as np
 
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import MarianMTModel, MarianTokenizer
 
 model_name = "Helsinki-NLP/opus-mt-en-ar"
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+
+# Use MarianTokenizer specifically
+tokenizer = MarianTokenizer.from_pretrained(model_name)
+model = MarianMTModel.from_pretrained(model_name)
 
 def translate_to_arabic(text):
     inputs = tokenizer(text, return_tensors="pt", padding=True)