Hey @Sona Kapoor thank you for reading my article and going through the code.

I hope you are doing well.

Regarding your error, so far my investigations point to you using a image that is not in RGB format.

Are you using colored RGB images ?