Update README.md
#54
by
staxjp
- opened
It requires torch to be imported as well.
The readme was missing a required library "torch" with the pip install command and was added to the python example. Tested and working