Views
No views yet
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2
3text = "You are a smart robot"
4fin_model_name = "fin-model-en-v0.1"
5
6fin_tokenizer = AutoTokenizer.from_pretrained(fin_model_name)
7fin_model = AutoModelForSequenceClassification.from_pretrained(fin_model_name)
8
9fin_inputs = fin_tokenizer(text, return_tensors="pt", padding="longest", truncation=True)
10fin_outputs = fin_model(**fin_inputs)
11fin_logits = fin_outputs.logits.squeeze(-1).float().detach().numpy()
12
13fin_score = fin_logits.item()
14result = {
15 "text": text,
16 "fin_score": fin_score,
17 "fin_int_score": int(round(max(0, min(fin_score, 5))))
18}
19
20print(result)
21# {'text': 'You are a smart robot', 'fin_score': 0.3258197605609894, 'fin_int_score': 0}