GenAI_project / scripts /deploy_model.py
jaothan's picture
Upload 24 files
fa64206 verified
raw
history blame contribute delete
484 Bytes
from huggingface_hub import HfApi
import os
def deploy_model_to_huggingface(model_path, model_name):
api = HfApi()
api.upload_folder(
folder_path=model_path,
repo_id=model_name,
commit_message="Deploying fine-tuned model",
use_auth_token=os.getenv("HF_AUTH_TOKEN")
)
if __name__ == "__main__":
model_path = "./results_student"
model_name = "your_model_name"
deploy_model_to_huggingface(model_path, model_name)