diff --git a/evaluation.py b/evaluation.py
index b0673b5..12ac135 100644
--- a/evaluation.py
+++ b/evaluation.py
@@ -5,7 +5,7 @@ import matplotlib.pyplot as plt
 from keras import backend as K
 from sklearn.metrics import mean_squared_error
 
-model = keras.models.load_model('suicide_model')
+model = keras.models.load_model('suicide_model.h5')
 
 # train = pd.read_csv('train.csv')
 # test = pd.read_csv('test.csv')
diff --git a/training.py b/training.py
index ac0ba8e..8ecbbb1 100644
--- a/training.py
+++ b/training.py
@@ -91,4 +91,4 @@ test_predictions = model.predict(X_test).flatten()
 predictions = model.predict(X_test)
 pd.DataFrame(predictions).to_csv('results.csv')
 
-model.save('suicide_model')
+model.save('suicide_model.h5')