Update app.py

gradcam/app.py

@@ -30,32 +30,31 @@ def grad_cam_fn(text, img, saliency_layer):
     return attn_map
 
 
-if __name__ == '__main__':
-    interface = gr.Interface(
-        fn=grad_cam_fn,
-        inputs=[
-            gr.inputs.Textbox(
-                label="Target Text",
-                lines=1),
-            gr.inputs.Image(
-                label='Input Image',
-                image_mode="RGB",
-                type='pil',
-                shape=(512, 512)),
-            gr.inputs.Dropdown(
-                ["layer4", "layer3", "layer2", "layer1"],
-                default="layer4",
-                label="Saliency Layer")
-        ],
-        outputs=gr.outputs.Image(
-            type="pil", 
-            label="Attention Map"),
-        examples=[
-            ['a cat lying on the floor', 'assets/cat_dog.jpg', 'layer4'],
-            ['a dog sitting', 'assets/cat_dog.jpg', 'layer4']
-        ],
-        description="OpenAI CLIP Grad CAM")
-    interface.launch(
-        server_name='0.0.0.0',
-        server_port=7861,
-        share=False)
+interface = gr.Interface(
+    fn=grad_cam_fn,
+    inputs=[
+        gr.inputs.Textbox(
+            label="Target Text",
+            lines=1),
+        gr.inputs.Image(
+            label='Input Image',
+            image_mode="RGB",
+            type='pil',
+            shape=(512, 512)),
+        gr.inputs.Dropdown(
+            ["layer4", "layer3", "layer2", "layer1"],
+            default="layer4",
+            label="Saliency Layer")
+    ],
+    outputs=gr.outputs.Image(
+        type="pil", 
+        label="Attention Map"),
+    examples=[
+        ['a cat lying on the floor', 'assets/cat_dog.jpg', 'layer4'],
+        ['a dog sitting', 'assets/cat_dog.jpg', 'layer4']
+    ],
+    description="OpenAI CLIP Grad CAM")
+interface.launch(
+    server_name='0.0.0.0',
+    server_port=7861,
+    share=False)