File size: 315 Bytes
2b63102 8f6bb93 2b63102 8f6bb93 2b63102 | 1 2 3 4 5 6 7 8 9 10 11 | """Download the embedding and NLI models at build time so the container runs offline."""
from src.embeddings import get_sentence_transformer
from src.evaluator import _nli
if __name__ == "__main__":
print("Prefetching embedding and NLI models...")
get_sentence_transformer()
_nli()
print("Done.")
|