implmented same fix as gradecam to test it out locally