That second model should definitely work. What platform are you on? Are you running this in a notebook or as a script?