Spaces:
Sleeping
Sleeping
Update train_model.py
Browse files- 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
|