Add main

app.py

@@ -97,17 +97,18 @@ def inference(img):
 
     return resnet_label, cam_result, vit_label, rollout_result
 
-
-interface = gr.Interface(
-    fn=inference,
-    inputs=gr.inputs.Image(type="pil", label="Input Image"),
-    outputs=[
-        gr.outputs.Label(num_top_classes=1, type="auto", label="ResNet Label"),
-        gr.outputs.Image(type="auto", label="ResNet CAM"),
-        gr.outputs.Label(num_top_classes=1, type="auto", label="ViT Label"),
-        gr.outputs.Image(type="auto", label="Rollout Attn Flow"),
-    ],
-    title="CNN - Transformer Explainability",
-    live=True,
-)
-interface.launch()
+if __name__ == "__main__":
+    interface = gr.Interface(
+        fn=inference,
+        inputs=gr.inputs.Image(type="pil", label="Input Image"),
+        outputs=[
+            gr.outputs.Label(num_top_classes=1, type="auto", label="ResNet Label"),
+            gr.outputs.Image(type="auto", label="ResNet CAM"),
+            gr.outputs.Label(num_top_classes=1, type="auto", label="ViT Label"),
+            gr.outputs.Image(type="auto", label="Rollout Attn Flow"),
+        ],
+        examples=examples,
+        title="CNN - Transformer Explainability",
+        live=True,
+    )
+    interface.launch()