Update app.py

app.py

@@ -5,14 +5,15 @@ from PIL import Image
 import os
 
 # --- مدل‌ها ---
-model_name = "persiannlp/mt5-base-parsinlu-opus-translation"
-
 try:
+    model_name = "persiannlp/mt5-small-parsinlu-opus-translation"
+    
     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)}")
+    raise gr.Error(f"خطا در بارگذاری مدل‌ها: لطفاً از موجود بودن مدل اطمینان حاصل کنید. خطا: {str(e)}")
 
 # --- توابع پردازش ---
 def run_ocr(image):