import numpy as np
from transformers import AutoModel

# Load sm_subgroup_classifier
sm_classifier = AutoModel.from_pretrained(
    "erikhenriksson/sm-subgroup-classifier", trust_remote_code=True
)

# create a random 1024 dimensional embedding
embedding = np.random.rand(1024).astype(np.float32)

# Use - model automatically discovers what's available
result = sm_classifier("fi", "NA-nb", embedding)

print(result)