Add integration tests for prediction

Since we will be redoing the code, we need an integration test that will be run each time the code changes. This should load a pre-trained model and check that the predictions on a few images are as expected.