Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tok = AutoTokenizer.from_pretrained("CyberKunju/finehance-categorizer-minilm")
5model = AutoModelForSequenceClassification.from_pretrained("CyberKunju/finehance-categorizer-minilm")
6model.eval()
7
8desc = "UBER *EATS HELP.UBER.COM"
9inputs = tok(desc, return_tensors="pt", truncation=True, max_length=64)
10with torch.no_grad():
11 logits = model(**inputs).logits
12pred_id = logits.argmax(-1).item()
13print(model.config.id2label[pred_id])