Update app.py

app.py

@@ -151,8 +151,8 @@ def get_grad(img):
 
 demo = gr.Interface(
     fn = get_grad,
-    inputs = gr.Image(type = "pil", shape = (224,224), width = 320, height = 320),
-    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]')], 
+    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]')], 
     description = "Visual Explanations from Deep Networks",
     title = "Gradient-Weighted Class Activation Mapping (Grad-CAM)"
 )