from evo_model import EvoTransformerConfig, EvoTransformerForClassification
import os

def initialize_and_save_model():
    config = EvoTransformerConfig()
    model = EvoTransformerForClassification(config)
    os.makedirs("trained_model", exist_ok=True)
    model.save_pretrained("trained_model")
    print("✅ EvoTransformer initial model saved to 'trained_model/'")

if __name__ == "__main__":
    initialize_and_save_model()