Update api.py

api.py

@@ -275,7 +275,7 @@ async def classify(image: UploadFile = File(...)):
         predicted_class_idx = int(predicted_class_idx)
 
         heatmap = make_gradcam_heatmap(img_array, model, last_conv_layer_name)
-        base64_image = display_gradcam(img_array, heatmap)
+        base64_image = display_gradcam( np.array(img), heatmap)
         
         # Return the base64 encoded image in the response