# save_model.py
from model.luna_model import LunaAI
from transformers import BertTokenizer

def save_model(model, path='./luna_ai_model'):
    model.save_pretrained(path)
    tokenizer = BertTokenizer.from_pretrained('bert-base-uncased')
    tokenizer.save_pretrained(path)

if __name__ == "__main__":
    model = LunaAI(num_classes=2)  # Adjust num_classes if necessary
    save_model(model)