Update init_model.py

init_model.py

@@ -19,5 +19,10 @@ def initialize_and_save_model():
 
     print("✅ EvoTransformer and tokenizer initialized and saved to 'trained_model/'")
 
+def load_model():
+    model = EvoTransformerForClassification.from_pretrained("trained_model")
+    return model
+
+# Optional: reinitialize if run directly
 if __name__ == "__main__":
     initialize_and_save_model()