predict now returns a numpy array