Hugging Face's logo

Spaces:
WwYc
/
explain-ViT
Sleeping

App Files Community
WwYc commited on
Commit
ade6eb5
·
verified ·
1 Parent(s): cb60283

Update visualization.py

Browse files
Files changed (1) hide show
  1. visualization.py +1 -1
visualization.py CHANGED
@@ -29,7 +29,7 @@ def show_cam_on_image(img, mask):
29
  return cam
30
 
31
  # initialize ViT pretrained
32
- model = vit_LRP(pretrained=True).cuda()
33
  model.eval()
34
  attribution_generator = LRP(model)
35
 
 
29
  return cam
30
 
31
  # initialize ViT pretrained
32
+ model = vit_LRP(pretrained=True)
33
  model.eval()
34
  attribution_generator = LRP(model)
35