Sorry for the inconvenience @Musa Would you be able to open a github issue on your finding and try try this Google Colab meanwhile?: https://colab.research.google.com/drive/1WjBbQzaivQhOldGolzymOtLmo6QX4Ieg#scrollTo=HXKT2-fpUD0Z