embeddings.py

import spaces
from sentence_transformers import SentenceTransformer

model = None

@spaces.GPU
def get_embeddings(texts):
    global model

    if model is None:
        model = SentenceTransformer("all-MiniLM-L6-v2")

    return model.encode(texts)