Error with pretrained model: 'dict' object has no attribute 'to_json_string'

Hey, works on Collab :slight_smile: