fix imports to work with transformers 0.4
Merge pull request #5 from ibotty/transformers-0.4

fix imports to work with transformers 0.4