shaqaqio/Qwen3-0.6B-steps-classifier

How to Use the model

import importlib.util
import torch

repo_path = "your/path/to/Qwen3-0.6B-steps-classifier"
model_py = f"{repo_path}/steps_classifier.py"

spec = importlib.util.spec_from_file_location("steps_model", model_py)
module = importlib.util.module_from_spec(spec)
spec.loader.exec_module(module)

MyModel = module.LegalSteps_Classifier

tokenizer = AutoTokenizer.from_pretrained(
    module.STEPS_ENCODER_NAME,
    trust_remote_code=True,
    use_fast=True
)

model = MyModel(
    encoder_name=module.STEPS_ENCODER_NAME,
    embedding_dim=module.EMBED_DIM,
    num_classes=module.NUM_CLASSES,
    hidden_dims=module.HIDDEN_DIMS,
    dropout=module.DROPOUT,
    trust_remote_code=True
)
model.load_state_dict(torch.load(f"{repo_path}/qwen3-legal-0.6B-classifier-new_best_model.pth"))
model.eval()