debug

app.py

@@ -26,6 +26,8 @@ METHOD_MAP = {
 
 
 def generate_viz(image, method, class_index=None):
+    if class_index is not None:
+        class_index = int(class_index)
     print(f"Image: {image.size}")
     print(f"Method: {method}")
     print(f"Class: {class_index}")
@@ -58,7 +60,7 @@ iface = gr.Interface(
         ["Transformer-Explainability/samples/catdog.png", "tiba", None],
         ["Transformer-Explainability/samples/catdog.png", "rollout", 243],
         ["Transformer-Explainability/samples/el2.png", "tiba", None],
-        # ["Transformer-Explainability/samples/el2.png", "gradcam", 340],
+        ["Transformer-Explainability/samples/el2.png", "gradcam", 340],
         ["Transformer-Explainability/samples/dogbird.png", "lrp", 161],
     ],
 )