The second code part uses the pretrained model, which should be faster.
Anyway, could you explain a little what you have done so far? As far as I understand you have already trained a model and would like to test it. Does this model have 4 input channels? What is the error in testing?