# Example functions def get_text_features(text): inputs = tokenizer(text, return_tensors="pt") outputs = text_model(**inputs) return outputs.last_hidden_state[:, 0, :] # Get the CLS token features