Modifying the HelloWorld component to support image classification