# Example: Load and Use the Model

import torch
import json
from huggingface_hub import hf_hub_download

# Your repository ID
repo_id = "YOUR_USERNAME/YOUR_REPO_NAME"  # Update this!

# Download files
config_path = hf_hub_download(repo_id=repo_id, filename="model_config.json")
weights_path = hf_hub_download(repo_id=repo_id, filename="model_weights.pt")
tokenizer_path = hf_hub_download(repo_id=repo_id, filename="tokenizer.json")

print("Files downloaded successfully!")

# Load and use your model
# (Add your TransformerModel class here)