# ==============================
# 上傳fine tuned後的BERT模型
# ==============================
import os
from huggingface_hub import HfApi,HfFolder
# 登入 Hugging Face
hf_token = os.environ["TOGETHER_API_KEY"]

HfFolder.save_token(hf_token)
api = HfApi(token=os.getenv("HF_TOKEN"))
output_dir = "./albert-tiny-chinese-finetuned"
api.upload_folder(
    folder_path=output_dir,
    repo_id="picard47at/tuned-albert-tiny",
    repo_type="model",
)