Update app.py

app.py

@@ -91,7 +91,7 @@ def create_heatmap(model, imgs, class_index):
 
 
 #superimpose function buth for a single input image
-def superimpose_single(heatmap, img, alpha = 0.7):
+def superimpose_single(heatmap, img, original_img_shape, alpha = 0.7):
     heatmap = np.uint8(255 * heatmap)
     
     # Use jet colormap to colorize heatmap