class_docstr: test show heatmap for ViT.
class_docstr_tok: ["test","show","heatmap","for","ViT","."]
code: def setUp(self) -> None:
      self.model = vit.create_model(
          model_name="vit_base_16x224", pretrained=False
      )
  
file_path: towhee/tests/unittests/models/visualization/test_transformer_vis.py
imports: import unittest
  from pathlib import Path
  from towhee.utils.pil_utils import PILImage
  from towhee.utils.matplotlib_utils import matplotlib
  from towhee.models.visualization.transformer_visualization import show_image_heatmap
  from towhee.models import vit
  from towhee.models.multiscale_vision_transformers import create_mvit_model
label_desc: Use image node_id 1 for calling the TestViTShow obj's underlying member method code with example usage: obj.setUp() without return types