Update app.py

app.py

@@ -16,6 +16,17 @@ os.environ["HOME"] = safe_home
 cfg_dir = os.path.join(safe_home, ".streamlit")
 os.makedirs(cfg_dir, exist_ok=True)
 
+
+# --- قبل از import streamlit، احیاناً مسیر کش قابل‌نوشتن:
+import os, tempfile
+os.environ.setdefault("HF_HOME", "/tmp/hf_home")
+os.makedirs(os.environ["HF_HOME"], exist_ok=True)
+import tempfile, os
+temp_dir = os.path.join(tempfile.gettempdir(), "trustocr_temp")
+os.makedirs(temp_dir, exist_ok=True)
+# جای "temp_files" استفاده کن
+
+
 # اطمینان از اینکه Streamlit همه فایل‌ها را اینجا می‌نویسد
 os.environ["STREAMLIT_CONFIG_DIR"] = cfg_dir
 
@@ -173,15 +184,30 @@ col2, col1 = st.columns([.5, .5])
 def load_det_cached():
     return load_det_model(checkpoint="vikp/surya_det2"), load_det_processor(checkpoint="vikp/surya_det2")
 
-from huggingface_hub import HfFolder
-HF_TOKEN = os.environ.get("HF_TOKEN")
+# from huggingface_hub import HfFolder
+# HF_TOKEN = os.environ.get("HF_TOKEN")
+
+# @st.cache_resource(show_spinner=True)
+# def load_rec_cached():
+#     return load_rec_model(checkpoint="MohammadReza-Halakoo/TrustOCR", token=HF_TOKEN), \
+#            load_rec_processor(checkpoint="MohammadReza-Halakoo/TrustOCR", token=HF_TOKEN)
 
 @st.cache_resource(show_spinner=True)
 def load_rec_cached():
-    return load_rec_model(checkpoint="MohammadReza-Halakoo/TrustOCR", token=HF_TOKEN), \
-           load_rec_processor(checkpoint="MohammadReza-Halakoo/TrustOCR", token=HF_TOKEN)
-
-
+    checkpoints = [
+        "MohammadReza-Halakoo/TrustOCR",  # خصوصی
+        "vikp/surya_rec2",                # عمومی (fallback)
+    ]
+    last_err = None
+    for ckpt in checkpoints:
+        try:
+            m = load_rec_model(checkpoint=ckpt)
+            p = load_rec_processor(checkpoint=ckpt)
+            return m, p
+        except Exception as e:
+            last_err = e
+    st.error(f"Loading recognition checkpoint failed: {last_err}")
+    raise last_err
 # @st.cache_resource(show_spinner=True)
 # def load_rec_cached():
     # return load_rec_model(checkpoint="MohammadReza-Halakoo/TrustOCR"), \