| from huggingface_hub import HfApi, login | |
| import os | |
| import time | |
| import sys | |
| def upload_to_huggingface(repo_name, wait_minutes): | |
| # Convert minutes to seconds and wait | |
| print(f"Script will upload files in {wait_minutes} minutes...") | |
| time.sleep(wait_minutes * 60) | |
| # Initialize Hugging Face API | |
| api = HfApi() | |
| try: | |
| # Authenticate using your Hugging Face API token | |
| hf_token = os.getenv("HF_API_TOKEN") | |
| if not hf_token: | |
| raise ValueError("Hugging Face API token is not set. Please set it in the environment variable 'HF_API_TOKEN'.") | |
| login(token=hf_token) | |
| api.create_repo(repo_id=repo_name, private=True, exist_ok=True) | |
| # Get current directory | |
| current_dir = os.getcwd() | |
| print("Starting upload to Hugging Face...") | |
| # Upload all files from current directory | |
| api.upload_folder( | |
| folder_path=current_dir, | |
| repo_id=repo_name, | |
| repo_type="model" | |
| ) | |
| print("Upload completed successfully!") | |
| except Exception as e: | |
| print(f"An error occurred: {str(e)}") | |
| if __name__ == "__main__": | |
| if len(sys.argv) != 3: | |
| print("Usage: python script.py <repository_name> <minutes_to_wait>") | |
| print("Example: python script.py 'your-username/model-name' 120") | |
| sys.exit(1) | |
| repo_name = sys.argv[1] | |
| wait_minutes = int(sys.argv[2]) | |
| upload_to_huggingface(repo_name, wait_minutes) | |