#!/usr/bin/env python3
from transformers import AutoModelForCausalLM, AutoTokenizer

# HuggingFace credentials
HF_TOKEN = "hf_"
REPO_ID = "ucalyptus/prem-1B-SQL-grpo"

# Local model path (using latest checkpoint)
MODEL_PATH = "outputs/Prem-1B-SQL-GRPO-v2/checkpoint-295"

def upload_model():
    print(f"Loading model from {MODEL_PATH}")
    
    # Load model and tokenizer
    model = AutoModelForCausalLM.from_pretrained(MODEL_PATH)
    tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
    
    print(f"Uploading model to {REPO_ID}")
    
    # Push to hub
    model.push_to_hub(
        REPO_ID,
        token=HF_TOKEN,
        private=False
    )
    
    tokenizer.push_to_hub(
        REPO_ID,
        token=HF_TOKEN,
        private=False
    )
    
    print("Upload complete!")

if __name__ == "__main__":
    upload_model()