| # save_model.py | |
| from model.luna_model import LunaAI | |
| from transformers import BertTokenizer | |
| def save_model(model): | |
| model.save_pretrained('./luna_ai_model') | |
| tokenizer = BertTokenizer.from_pretrained('bert-base-uncased') | |
| tokenizer.save_pretrained('./luna_ai_model') | |
| if __name__ == "__main__": | |
| model = LunaAI() | |
| save_model(model) |