Spaces:

OmidSakaki
/

DocQA_Agent

Sleeping

OmidSakaki commited on Jul 2

Commit

999a6b4

verified ·

1 Parent(s): e186575

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -6,8 +6,7 @@ import os
 # --- مدل‌ها ---
 try:
-    # مدل جدید تست شده و کارآمد
-    model_name = "m3hrdadfi/mt5-small-parsinlu-grammar-correction"
     ocr_model = PaddleOCR(lang='fa', use_textline_orientation=True)
     tokenizer = AutoTokenizer.from_pretrained(model_name)

 # --- مدل‌ها ---
 try:
+    model_name = "m3hrdadfi/mt5-small-finetuned-grammar-synthesis"
     ocr_model = PaddleOCR(lang='fa', use_textline_orientation=True)
     tokenizer = AutoTokenizer.from_pretrained(model_name)