when run the test code, it seems that we need the pre-trained model. Will you release the pre-trained model?