Spaces:

all-things-vits
/

comparative-explainability

Sleeping

sayakpaul HF staff commited on Jun 12, 2023

Commit

7d808f1

•

1 Parent(s): 26b53b0

setting explicit argument

Files changed (1) hide show

app.py CHANGED Viewed

@@ -27,7 +27,7 @@ METHOD_MAP = {
 def generate_viz(image, method, class_index=None):
     viz_method = METHOD_MAP[method]
-    viz = viz_method(TRANSFORM, image, class_index)
     viz.savefig("visualization.png")
     return Image.open("visualization.png").convert("RGB")

 def generate_viz(image, method, class_index=None):
     viz_method = METHOD_MAP[method]
+    viz = viz_method(TRANSFORM, image, class_index=class_index)
     viz.savefig("visualization.png")
     return Image.open("visualization.png").convert("RGB")