scripts/download_model.py

import os
import sys

from sentence_transformers import SentenceTransformer

# sys.path.insert(0, os.path.join(os.path.dirname(__file__), ".."))
# from src.book_recommender.core import config

MODEL_NAME = "all-MiniLM-L6-v2"

def download_model():
    """
    Downloads the sentence-transformer model specified in the config file.
    This is useful for pre-downloading the model in a Docker build.
    """
    print(f"Downloading model: {MODEL_NAME}...")
    _ = SentenceTransformer(MODEL_NAME)
    print("Model downloaded successfully.")


if __name__ == "__main__":
    download_model()