Updated path to load learned model