Appreciate it! Looking forward to seeing it in real world situations indeed!

I'll take a look and update as necessary. Thank you for letting me know.

Thank you for the kind words. With Colab, you may need to restart the runtime. This happens when pip install updates an existing package. I was able to successfully run the notebook on Colab. If you continue to encounter problems, please submit an issue on GitHub - https://github.com/neuml/txtai.