# 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)