Update app.py

app.py

@@ -17,7 +17,7 @@ class OCRProcessor:
 
 class TextCorrector:
     def __init__(self):
-        model_name = "HooshvareLab/mt5-small-fa"
+        model_name = "HooshvareLab/mt5-small-parsbert-uncased"
         try:
             self.tokenizer = AutoTokenizer.from_pretrained(model_name)
             self.model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
@@ -29,7 +29,7 @@ class TextCorrector:
             return text
             
         try:
-            prompt = "تصحیح نگارشی متن: " + text
+            prompt = "بازنویسی و تصحیح: " + text
             inputs = self.tokenizer(
                 prompt,
                 return_tensors="pt",