Update app.py

app.py

@@ -38,7 +38,7 @@ gr.Interface(
                                                                               get_misclassified_images(show_misclassified, num_images) if show_misclassified else None,
                                                                               get_gradcam(model,image, opacity,layer) if show_gradcam else None],
     inputs=[input_component, num_of_output_classes,show_misclassified_checkbox, num_images_input, show_gradcam_checkbox, opacity_slider,layer_input],
-    outputs=[gr.outputs.Label(), gr.Image(shape=(500, 500)), gr.Image(shape=(500, 500))],
+    outputs=[gr.outputs.Label(), gr.Image(shape=(500, 500)), gr.Image(shape=(250, 250))],
     title="CIFAR10 Trained on Custom Residual CNN Architecture",
     examples=[
         ["example_images/example_1.png",5,True,5,True,0.2,'layer3'],  # You can provide your own example input values here