Update BERTmodel.py

BERTmodel.py

@@ -18,6 +18,7 @@ CACHE_DIR  = "./hf_cache"
 class BERTVisualizer(TransformerVisualizer):
     def __init__(self,task):
         super().__init__()  
+        print(task,'BERTVIS START')
         self.task = task
         self.tokenizer = BertTokenizer.from_pretrained('bert-base-uncased', cache_dir=CACHE_DIR)
         print('finding model', self.task)