Update app.py

app.py

@@ -88,7 +88,7 @@ examples = [["cat.jpg", 1, -1, 0.8, True, 3, 3],
            ]
 demo = gr.Interface(
     inference,
-    inputs=[gr.Image(label="Input Image"),
+    inputs=[gr.Image(width=256, height=256, label="Input Image"),
             gr.Slider(1, 10, value=1, step=1, label="Number of GradCAM Images"),
             gr.Slider(-2, -1, value=-1, step=1, label="Which Layer?"),
             gr.Slider(0, 1, value=0.8, label="Opacity of GradCAM"),
@@ -96,8 +96,8 @@ demo = gr.Interface(
             gr.Slider(2, 10, value=3, step=1, label="Top Predictions"),
             gr.Slider(1, 10, value=3, step=1, label="Misclassified Images")],
     outputs=[gr.Label(label="Top Predictions"),
-             gr.Image(label="Output").style(width=128, height=128),
-             gr.Image(label="Misclassified Images").style(width=640, height=360)],
+             gr.Image(label="Output",width=128, height=128),
+             gr.Image(label="Misclassified Images",width=640, height=360)],
     title=title,
     description=description,
     examples=examples,