Update app.py

app.py

@@ -79,6 +79,7 @@ def predict(img):
     tumor_present = pred.max() > 0.7
     
     if tumor_present:
+        mask = (pred > 0.7).astype(np.uint8) * 255
         mask = remove_small_blobs(mask, min_size=50)
         mask_img = Image.fromarray(mask.squeeze()).convert("L")