Update app.py

app.py

@@ -5,9 +5,14 @@ from PIL import Image
 import os
 
 # --- مدل‌ها ---
-ocr_model = PaddleOCR(lang='fa', use_textline_orientation=True)
-tokenizer = AutoTokenizer.from_pretrained("persiannlp/mt5-small-parsinlu-grammar-correction")
-nlp_model = AutoModelForSeq2SeqLM.from_pretrained("persiannlp/mt5-small-parsinlu-grammar-correction")
+model_name = "persiannlp/mt5-base-parsinlu-opus-translation"
+
+try:
+    ocr_model = PaddleOCR(lang='fa', use_textline_orientation=True)
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    nlp_model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+except Exception as e:
+    raise gr.Error(f"خطا در بارگذاری مدل‌ها: {str(e)}")
 
 # --- توابع پردازش ---
 def run_ocr(image):