ok Will check your codes, get back. seems like image tensor are of different shape.
@Alavudeen_Ahamed I am going to respond to your created post just to avoid confusion with the post creators post as this issue was totally related dependency version issue.