It seems that the model testing part is missing from the existing code. Can you provide the code of this part?