s

train.py

@@ -72,7 +72,7 @@ def train_model(data_file, model_file, epochs, batch_size, test_size, random_sta
     print('Test accuracy:', accuracy)
     print('Test loss:', loss)
 
-    model.save(model_file)
+    model.save("model.h5")
     input_signature = {
         'input': tensor_spec.TensorSpec(shape=X_train[0].shape, dtype=X_train.dtype)
     }