Index out of range in self when I predict all tokens using BERT

Question

When I try to predict all tokens in my text using BERT with this function:

        def load_predict_BERT(self, masked_text