diff --git a/predict.py b/predict.py
index 290b71f..ccba156 100644
--- a/predict.py
+++ b/predict.py
@@ -7,7 +7,7 @@ from sklearn.model_selection import train_test_split
 import tensorflow
 
 
-reloaded = tensorflow.keras.models.load_model('.')
+reloaded = tensorflow.keras.models.load_model('saved_model.pb')
 
 x_to_test = pandas.read_csv('./X_test.csv')
 y_to_test = pandas.read_csv('./Y_test.csv')