diff --git a/predict.py b/predict.py
index af76db4..aa6060c 100644
--- a/predict.py
+++ b/predict.py
@@ -9,8 +9,12 @@ from keras.metrics import MeanSquaredError
 
 loaded_model = tf.keras.models.load_model('powerlifting_model.h5')
 
-data = pd.read_csv('openpowerlifting.csv')
+data = pd.read_csv('./data/train.csv')
+
 data = data[['Sex', 'Age', 'BodyweightKg', 'TotalKg']].dropna()
+data['Age'] = pd.to_numeric(data['Age'], errors='coerce')
+data['BodyweightKg'] = pd.to_numeric(data['BodyweightKg'], errors='coerce')
+data['TotalKg'] = pd.to_numeric(data['TotalKg'], errors='coerce')
 features = data[['Sex', 'Age', 'BodyweightKg']]
 target = data['TotalKg']