Spaces:

probing-vits
/

class-saliency

Runtime error

sayakpaul HF staff commited on May 3, 2022

Commit

486cd38

1 Parent(s): 7f209a3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -46,7 +46,7 @@ def show_plot(image):
     heatmap = np.float32(heatmap)
     saliency_map = heatmap + original_image
-    saliency_map = np.clip(saliency_map, 0.0, 255.0).astype("int")
     return Image.fromarray(saliency_map)

     heatmap = np.float32(heatmap)
     saliency_map = heatmap + original_image
+    saliency_map = np.clip(saliency_map, 0.0, 255.0).astype("int32")
     return Image.fromarray(saliency_map)