I'm actually using custom data to train the model and I want to use the test.py on my test split. Anyone has any idea how to do it ?