3. Image recognition model python implementation code