shaqaqio/Qwen3-4B-verdict-classifier

How to Use the Model

import importlib.util
import torch

repo_path = "your/path/to/Qwen3-4B-verdict-classifier"
model_py = f"{repo_path}/verdict_classifier.py"

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

MyModel = module.LegalVerdict_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-4B-classifier-verdict_best_model.pth"))
model.eval()