embeddings.py

from sklearn.feature_extraction.text import TfidfVectorizer

vectorizer = TfidfVectorizer(max_features=512)

def embed(text):

    vec = vectorizer.fit_transform([text])

    return vec.toarray()[0].tolist()