Update init_model.py

init_model.py

@@ -1,18 +1,12 @@
-import torch
-import os
 from evo_model import EvoTransformerConfig, EvoTransformerForClassification
+import os
 
-def initialize_evo_model():
-    print("⚙️ Reinitializing EvoTransformer model...")
-    
-    # Create default config
+def initialize_and_save_model():
     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/'")
+
+if __name__ == "__main__":
+    initialize_and_save_model()