diff --git a/train.py b/train.py index 11c3c51..0bf4350 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("app/savedmodel") +model.save("savedmodel")