upload.py

from huggingface_hub import login
login()  # ����ʾ������token

from huggingface_hub import HfApi, create_repo
import os

# ���������Ϣ
username = "����û���"
model_name = "reward-model-qwen"
repo_id = f"{username}/{model_name}"

# ����APIʵ��
api = HfApi()

# �����ֿ�
try:
    create_repo(repo_id, repo_type="model", private=False)
    print(f"�����ֿ�: {repo_id}")
except Exception as e:
    print(f"�ֿ��Ѵ��ڻ򴴽�ʧ��: {e}")

# �ϴ�ģ���ļ�
model_path = "./reward_model_output/final_model"
if os.path.exists(model_path):
    api.upload_folder(
        folder_path=model_path,
        repo_id=repo_id,
        repo_type="model",
        commit_message="�ϴ�ѵ���õĽ���ģ�� v1.0"
    )
    print(f"ģ�����ϴ���: https://huggingface.co/{repo_id}")
else:
    print(f"ģ��·��������: {model_path}")