Update app.py

app.py

@@ -199,7 +199,6 @@ def generate_colored_image(labels):
     for i in range(0,w):
         for j in range(0,h):
             new_mk[i][j] = colors[labels[i][j]]
-    # new_mk = new_mk / 255.0
     new_mk = new_mk.astype(np.uint8)
     return Image.fromarray(new_mk)
 
@@ -270,7 +269,7 @@ def segment_image(image):
 
     return image, legend, stats, superimposed_image
 
-def superimpose_images(image1, image2, alpha=0.2):
+def superimpose_images(image1, image):
 
     if image1.dtype != np.uint8:
             image1 = (image1 * 255).astype(np.uint8) if image1.max() <= 1 else image1.astype(np.uint8)
@@ -280,10 +279,13 @@ def superimpose_images(image1, image2, alpha=0.2):
     # Resize image1 to match image2's size
     image1 = image1.resize(image2.size)
 
-    # Blend the images
-    blended = Image.blend(image1, image2, alpha)
+    image_np = np.array(image1)
+    heatmap_np = np.array(image2)
 
-    return blended
+    superimposed_np = cv2.addWeighted(heatmap_np, 0.2, image_np, 1, 0)
+    superimposed_pil = Image.fromarray(superimposed_np)
+
+    return superimposed_pil
 
 
 model_path = os.path.join(model_dir, 'spaderesnet.pt')