Correct Classification

Efficient classification and inference of image file