You can check out this notebook to try it out.
<a href="https://colab.research.google.com/drive/12idzfrTSRtEx5Dam9rTQcGv2c7s11nmo?usp=sharing" rel="nofollow">https://colab.research.google.com/drive/12idzfrTSRtEx5Dam9rT...</a>