Fixed the TF/IDF model to use np.array