INNER CODE UNIT · Python

_load_model

scub-france/docling-Studio · embedding-service/main.py:29

async def _load_model() -> None:
    global model  # noqa: PLW0603
    logger.info("Loading sentence-transformers model '%s' …", MODEL_NAME)
    t0 = time.monotonic()
    model = SentenceTransformer(MODEL_NAME)
    elapsed = time.monotonic() - t0
    dim = model.get_sentence_embedding_dimension()
    logger.info("Model loaded in %.1fs — dimension=%d", elapsed, dim)


# -- Schemas -------------------------------------------------------------------


class EmbedRequest(BaseModel):
    texts: list[str] = Field(..., min_length=1, description="Texts to embed")


class EmbedResponse(BaseModel):

View source record →