Update app.py

app.py

@@ -54,15 +54,15 @@ def predict_and_explain(img):
     image_np = np.array(image_pil).astype(np.float32) / 255.0
     cam_image = show_cam_on_image(image_np, grayscale_cam, use_rgb=True)
 
-    cam_pil = Image.fromarray(cam_image)
+    cam_pil = Image.fromarray(cam_image).resize(img.size)  # Resize to match original
     return cam_pil, f"Prediction: {pred_name} (class index {pred_idx})"
 
 # --- Gradio UI ---
 demo = gr.Interface(
     fn=predict_and_explain,
-    inputs=gr.Image(type="pil", label="Upload Car Image", height=350),
+    inputs=gr.Image(type="pil", label="Upload Car Image"),
     outputs=[
-        gr.Image(type="pil", label="Grad-CAM++ Heatmap", height=350),
+        gr.Image(type="pil", label="Grad-CAM++ Heatmap"),
         gr.Text(label="Prediction")
     ],
     title="🚗 EfficientNetV2 Car Classifier + Grad-CAM++",