Update app.py

app.py

@@ -152,7 +152,7 @@ def get_grad(img):
 demo = gr.Interface(
     fn = get_grad,
     inputs = gr.Image(type = "pil", shape = (224,224)),
-    outputs = [gr.Image(type = "numpy", height = 320, label = "Grad_CAM w.r.t cat"), gr.Image(type = "numpy", height = 320, label = "Grad_CAM w.r.t dog"), gr.Textbox(label = 'Prediction', info = '[P of cat, P of dog]')], 
+    outputs = [gr.Image(type = "numpy", width = 320, height = 320, label = "Grad_CAM w.r.t cat"), gr.Image(type = "numpy", width = 320, height = 320, label = "Grad_CAM w.r.t dog"), gr.Textbox(label = 'Prediction', info = '[P of cat, P of dog]')], 
     description = "Visual Explanations from Deep Networks",
     title = "Gradient-Weighted Class Activation Mapping (Grad-CAM)"
 )