Update app.py

app.py

@@ -50,9 +50,10 @@ def predict(img):
     
     if tumor_present:
         mask = (pred > 0.5).astype(np.uint8) * 255
-        mask_img = Image.fromarray(mask.squeeze()).convert("RGB")
-        overlay = img.convert("RGB")  # convert original image to RGB
-        overlay.paste(mask_img, (0, 0), mask_img)
+        mask_img = Image.fromarray(mask.squeeze()).convert("L")  # single channel for mask
+        overlay = img.convert("RGB")
+        red_overlay = Image.new("RGB", mask_img.size, (255,0,0))
+        overlay.paste(red_overlay, (0,0), mask_img)
         return "Tumor Detected", overlay
     else:
         return "No Tumor Detected", None