Unfortunately, we still need some time to support in TFLite Task library.
As for the issue that tensorflow==2.5.0 runs slow, it may be due to incompatible cuda/cudnn version inside Colab so that it’s not running on GPU which we can’t control. Could you please run on your local station instead?