deepugaur commited on
Commit
d5f4666
·
verified ·
1 Parent(s): 6049c7f

Update train_model.py

Browse files
Files changed (1) hide show
  1. train_model.py +3 -0
train_model.py CHANGED
@@ -50,3 +50,6 @@ model.fit(X_train_pad, y_train, epochs=5, batch_size=32, validation_data=(X_test
50
  # Save the trained model
51
  model.save("deep_learning_model.h5")
52
  print("Model saved to deep_learning_model.h5")
 
 
 
 
50
  # Save the trained model
51
  model.save("deep_learning_model.h5")
52
  print("Model saved to deep_learning_model.h5")
53
+
54
+ if __name__ == "__main__":
55
+ train_model() # Ensure this calls the training function