Update app.py

app.py

@@ -58,7 +58,7 @@ gradio_app = gr.Interface(
     inference,
     inputs = [gr.Image(width=32, height=32, label="Input Image"), gr.Slider(0, 1, value = 0.5, label="Opacity of GradCAM"), gr.Slider(-2, -1, value = -2, step=1,
                                                                                                                                  label="Which Layer?")],
-    outputs = [gr.Label(num_top_classes=3), gr.Image(width=32, height=32, label="Output").style(width=128, height=128)],
+    outputs = [gr.Label(num_top_classes=3), gr.Image(width=32, height=32, label="Output")],
     title = title,
     description = description,
     examples = examples,