updates function signature to match with API change in image package
Merge pull request #16 from soumith/master

updates function signature to match with API change in image package