Views
No views yet
1from transformers import DistilBertTokenizer, DistilBertForSequenceClassification
2import torch
3
4# Load model
5tokenizer = DistilBertTokenizer.from_pretrained("franzzzzzzzzz/tpb-subjective-norm-nutrition")
6model = DistilBertForSequenceClassification.from_pretrained("franzzzzzzzzz/tpb-subjective-norm-nutrition")
7
8# Predict
9text = "I love healthy food, it's amazing!"
10inputs = tokenizer(text, return_tensors='pt', padding=True, truncation=True)
11outputs = model(**inputs)
12prediction = torch.argmax(outputs.logits, dim=1).item()
13
14# Convert 0-4 to 1-5 scale
15score = prediction + 1
16print(f"Score: {score}/5")@misc{tpb-ttm-nutrition-models,
author = {Your Name},
title = {franzzzzzzzzz/tpb-subjective-norm-nutrition},
year = {2026},
publisher = {Hugging Face},
howpublished = {\url{https://huggingface.co/franzzzzzzzzz/tpb-subjective-norm-nutrition}}
}