import joblib
from huggingface_hub import snapshot_download

# Download model
local_dir = snapshot_download("jruffle/umap_samples_4d")
model = joblib.load(f"{local_dir}/model.joblib")

# For classical models (PCA/UMAP):
# model contains: 'pca', 'umap', 'robust_scaler', 'gene_order'

# For TabPFN models (with UMAP reduction):
# model contains: 'tabpfn_embedding', 'umap_final', 'raw_embeddings', 'input_scaler', etc.
# Uses UMAP instead of PCA for non-linear dimension reduction of TabPFN embeddings