| """Upload model + dict + corpus lên HF model repo (Space tải về lúc khởi động). |
| |
| Repo cũ (v4): HF_TOKEN=hf_xxx python push_models_hf.py |
| Repo mới (v11) — KHÔNG đè repo cũ: |
| HF_TOKEN=hf_xxx HF_MODEL_REPO=sirimiri/vocr-models-v11 \\ |
| COMMIT_MSG="rec v11 (acc 0.225, -27% lỗi lặp) + det mobile 0.916 + dict 233 + ViLM corpus" \\ |
| python push_models_hf.py |
| """ |
| import os |
| from huggingface_hub import HfApi |
|
|
| REPO = os.getenv("HF_MODEL_REPO", "sirimiri/vocr-models") |
| MSG = os.getenv( |
| "COMMIT_MSG", |
| "Update fine-tuned det (hmean 0.916) + rec + dict 233 chars + ViLM corpus", |
| ) |
| api = HfApi(token=os.environ["HF_TOKEN"]) |
| api.create_repo(REPO, repo_type="model", exist_ok=True) |
| api.upload_folder( |
| repo_id=REPO, repo_type="model", folder_path=".", |
| allow_patterns=[ |
| "models/inference_det/*", "models/inference_rec/*", |
| "models/det_best_accuracy.pdparams", "models/rec_best_accuracy.pdparams", |
| "models/config_det.yml", "models/vi_PP-OCRv5_mobile_rec.yml", |
| "dict/vi_vietnam.txt", "lm_data/vi_corpus.txt", |
| ], |
| commit_message=MSG, |
| ) |
| print(f"✅ Đã upload lên https://huggingface.co/{REPO}") |
|
|
