Okay. So I guess the two output numbers represent classes. If they are in 0-1 you could compute the color as a weighted sum of both:

red = np.array([1, 0, 0])
blue = np.array([0, 0, 1])
color = pred[0] * red + pred[1] * blue

compute and pass that color for every point.