ttw / model_s.py
freemt
Update model_s.py
d66b767
"""Load model_s."""
# pylint: disable=invalid-name
from pathlib import Path
import joblib
# from sentence_transformers import SentenceTransformer
from alive_progress import alive_bar
file_loc = "radiobee/model_s.lzma"
file_loc = "model_s.lzma"
if not Path(file_loc).exists():
raise Exception(f"File {file_loc} does not exist.")
with alive_bar(1, title="Loading model_s, takes up to ~2 minutes...", length=3) as progress_bar:
model_s = joblib.load(file_loc)
progress_bar() # pylint: disable=not-callable