Lightning flash pretraining produces empty string while doing prediction

Hi all,
I have finetuned the python-lightning-flash module on custom dataset. I followed the instructions from this link. When I am loading the checkpoint and starting to predict it produces empty string. Has someone faced this issue? Need urgent help.