Views
No views yet
deepset/gbert-largesafetensors1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4model = AutoModelForSequenceClassification.from_pretrained("aveluth/author_regulatory_focus_classifier")
5tokenizer = AutoTokenizer.from_pretrained("aveluth/author_regulatory_focus_classifier")
6
7text = "Wir müssen sicherstellen, dass keine Fehler passieren. Sicherheit hat höchste Priorität."
8inputs = tokenizer(text, return_tensors="pt")
9outputs = model(**inputs)
10predicted_class = torch.argmax(outputs.logits).item()
11
12print("Predicted class:", "prevention" if predicted_class == 0 else "promotion")| Class | Description |
|---|---|
0 | Prevention-focused language |
1 | Promotion-focused language |
1@article{velutharambath2023prevention,
2 title={Prevention or Promotion? Predicting Author's Regulatory Focus},
3 author={Velutharambath, Aswathy and Sassenberg, Kai and Klinger, Roman},
4 journal={Northern European Journal of Language Technology},
5 volume={9},
6 number={1},
7 year={2023}
8}