samlookingsamsamsam

Hi Thanks for the great work, when I try to save the predicted image in test function I face issue can you please add the line of code