| import os | |
| from huggingface_hub import HfApi, HfFolder, Repository | |
| from transformers import AutoModel, AutoTokenizer | |
| # Set your Hugging Face credentials | |
| HUGGINGFACE_TOKEN = "your_huggingface_access_token" # Replace with your token | |
| MODEL_NAME = "your-model-name" # Name of the model on Hugging Face Hub (e.g., "your-username/your-model") | |
| # Save token for authentication | |
| HfFolder.save_token(HUGGINGFACE_TOKEN) | |
| # Load a model (Example: BERT) | |
| model = AutoModel.from_pretrained("bert-base-uncased") | |
| tokenizer = AutoTokenizer.from_pretrained("bert-base-uncased") | |
| # Save the model and tokenizer locally | |
| MODEL_DIR = "./my_model" | |
| model.save_pretrained(MODEL_DIR) | |
| tokenizer.save_pretrained(MODEL_DIR) | |
| # Upload model to Hugging Face Hub | |
| api = HfApi() | |
| api.create_repo(repo_id=MODEL_NAME, exist_ok=True) # Create repo if it doesn't exist | |
| # Clone and push the model | |
| repo = Repository(local_dir=MODEL_DIR, clone_from=MODEL_NAME) | |
| repo.git_add() | |
| repo.git_commit("Initial commit: Upload model") | |
| repo.git_push() | |
| print(f"Model pushed to Hugging Face: https://huggingface.co/{MODEL_NAME}") | |