Update app.py

app.py

@@ -43,11 +43,10 @@ def show_plot(image):
     heatmap = cv2.applyColorMap(
         np.uint8(255 * result_first_block), cv2.COLORMAP_CIVIDIS
     )
-    heatmap = np.float32(heatmap) / 255
+    heatmap = np.float32(heatmap)
 
-    original_image = original_image / 255.0
     saliency_map = heatmap + original_image
-    saliency_map = saliency_map / np.max(saliency_map)
+    saliency_map = np.clip(saliency_map, 0.0, 255.0).astype("int")
     return Image.fromarray(saliency_map)