diff --git a/train.py b/train.py index 0bf4350..11c3c51 100644 --- a/train.py +++ b/train.py @@ -49,5 +49,5 @@ model.compile( ) model.fit(X_train_scaled, y_train, batch_size=32, epochs=5, verbose=2) -model.save("savedmodel") +model.save("app/savedmodel")