Views
No views yet
1from transformers import AutoTokenizer, AutoModel
2import torch
3
4# Load model and tokenizer
5model_name = "AWCO/mdeberta-v3-base-narratives-classifier-hierarchical"
6tokenizer = AutoTokenizer.from_pretrained(model_name)
7model = AutoModel.from_pretrained(model_name)
8
9# Example usage for inference
10text = "Your text here..."
11inputs = tokenizer(text, return_tensors="pt", truncation=True, padding=True, max_length=512)
12
13with torch.no_grad():
14 outputs = model(**inputs)
15 # Process outputs according to your inference pipelineconfig.json: Model configurationmodel.safetensors: Model weights in SafeTensors formattokenizer.json: Tokenizer configurationlabel_mappings_hierarchical.json: Hierarchical label mappings for classification