For prediction on test data use this link. https://colab.research.google.com/drive/1K-ED0yvMsdciNo52rluauQBEAg-DBomC?usp=sharing update best_model_path = "sumitrsch/multiconer2_muril_large_bn"