Update app.py

app.py

@@ -157,5 +157,13 @@ demo = gr.Interface(
     description = "Visual Explanations from Deep Networks",
     title = "Gradient-Weighted Class Activation Mapping (Grad-CAM)"
 )
+
+with gr.Blocks() as demo:
+    with gr.Row().style(equal_height=True):
+        img_input = gr.Image(type = "pil", shape = (224,224), width = 320, height = 320)
+        img_output1 = gr.Image(type = "numpy", width = 320, height = 320, label = "Grad_CAM w.r.t cat")
+        img_output2 = gr.Image(type = "numpy", width = 320, height = 320, label = "Grad_CAM w.r.t dog")
+        
+
 demo.launch()