Views
No views yet
config.json):model.py to instantiate the architecture and load weights:1from safetensors.torch import load_file
2from model import create_model
3
4model = create_model(num_classes=16)
5state_dict = load_file("model.safetensors")
6model.load_state_dict(state_dict)
7model.eval()config.json contains id2label/label2id mappings..safetensors format for safer deserialization.