how to make a threshold for predicted image from uNet

Question

I\'am training uNet model for microscopic images , when predicting an image with code bellow

 image = cv2.imread(image_path)
mask = cv2.imread(mask)

img = im