Huggingface demo is broken

[Newtoniano]

Hello, I found your huggingface demo of this project very cool, but now it seems to be broken because of a numpy error. Please consider fixing this when you have time (should be an easy fix), and keep up the awesome work!

[hasibzunair]

Hi, thanks for the interest in this work. I tried a fix with leads to another error. This is a tensorflow version problem I think. Feel free to suggest pointers to fix or maybe even a PR!