Upload train_model.py

train_model.py

@@ -1,4 +1,11 @@
+# -*- coding: utf-8 -*-
+"""train_model.ipynb
 
+Automatically generated by Colab.
+
+Original file is located at
+    https://colab.research.google.com/drive/1R-i5F70PPmj1iBbLI_xkYtDWxwgf9YEw
+"""
 
 import numpy as np
 import pandas as pd
@@ -51,5 +58,3 @@ model.fit(X_train_pad, y_train, epochs=5, batch_size=32, validation_data=(X_test
 model.save("deep_learning_model.h5")
 print("Model saved to deep_learning_model.h5")
 
-if __name__ == "__main__":
-    train_model()  # Ensure this calls the training function