MARBLE SURFACE CLASSIFIER USING PYTORCH MODEL