init_model.py

import torch
import os
from evo_model import EvoTransformerConfig, EvoTransformerForClassification

def initialize_evo_model():
    print("⚙️ Reinitializing EvoTransformer model...")
    
    # Create default config
    config = EvoTransformerConfig()

    # Create model
    model = EvoTransformerForClassification(config)

    # Save model to disk
    os.makedirs("trained_model", exist_ok=True)
    model.save_pretrained("trained_model")
    
    print("✅ EvoTransformer initial model saved to 'trained_model/'")