Correcting the file paths and improving the prediction model.