Initial Colab Notebook Demonstrating Inference

[AlexanderMath]

Great work. It takes some time (1-2 hours) to get all dependencies right and use your model for inference. I made a notebook so anyone can use it for inference in ~ 5 min. Let me know if I broke anything, to make it easier the notebook uses zeroshot.

https://colab.research.google.com/drive/1gr7RyiIZtFXgi4789iO1u1jvkEleAOA7?usp=sharing

[AlexanderMath]

In the demo it takes 1 iteration a second and has 2-4% GPU utilization.

[mariocorradetti]

Thank you, this is very useful.
You have also a full implementation of the model, becuase i wanna train this network on a custom dataset using Colab.
Thank you in advance