Hugging Face's logo

Spaces:
andreped
/
vit-explainer
Runtime error

App Files Community
andreped commited on
Commit
1281282
·
1 Parent(s): 420e4cc

Scaled saliency

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -48,6 +48,7 @@ class WebUI:
48
  saliency = np.squeeze(np.moveaxis(saliency, 1, 3))
49
  saliency[saliency >= 0.05] = 0.05
50
  saliency[saliency <= -0.05] = -0.05
 
51
  return saliency
52
 
53
  def run(self):
 
48
  saliency = np.squeeze(np.moveaxis(saliency, 1, 3))
49
  saliency[saliency >= 0.05] = 0.05
50
  saliency[saliency <= -0.05] = -0.05
51
+ saliency /= np.amax(np.abs(saliency))
52
  return saliency
53
 
54
  def run(self):