diff --git a/predict.py b/predict.py
index de75e9b..1611065 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('test')
+reloaded = tensorflow.keras.models.load_model('./')
 
 x_to_test = pandas.read_csv('./X_test.csv')
 y_to_test = pandas.read_csv('./Y_test.csv')