updated readme
Browse files
README.md
CHANGED
|
@@ -35,7 +35,7 @@ import torch
|
|
| 35 |
model = AutoModelForSequenceClassification.from_pretrained("aveluth/author_regulatory_focus_classifier")
|
| 36 |
tokenizer = AutoTokenizer.from_pretrained("aveluth/author_regulatory_focus_classifier")
|
| 37 |
|
| 38 |
-
text = ""
|
| 39 |
inputs = tokenizer(text, return_tensors="pt")
|
| 40 |
outputs = model(**inputs)
|
| 41 |
predicted_class = torch.argmax(outputs.logits).item()
|
|
@@ -43,8 +43,6 @@ predicted_class = torch.argmax(outputs.logits).item()
|
|
| 43 |
print("Predicted class:", "prevention" if predicted_class == 0 else "promotion")
|
| 44 |
```
|
| 45 |
|
| 46 |
-
> Make sure to replace `"your-username/..."` with the correct model path.
|
| 47 |
-
|
| 48 |
## Labels
|
| 49 |
|
| 50 |
| Class | Description |
|
|
@@ -83,4 +81,4 @@ If you use this model in your research, please cite:
|
|
| 83 |
number={1},
|
| 84 |
year={2023}
|
| 85 |
}
|
| 86 |
-
```
|
|
|
|
| 35 |
model = AutoModelForSequenceClassification.from_pretrained("aveluth/author_regulatory_focus_classifier")
|
| 36 |
tokenizer = AutoTokenizer.from_pretrained("aveluth/author_regulatory_focus_classifier")
|
| 37 |
|
| 38 |
+
text = "Wir müssen sicherstellen, dass keine Fehler passieren. Sicherheit hat höchste Priorität."
|
| 39 |
inputs = tokenizer(text, return_tensors="pt")
|
| 40 |
outputs = model(**inputs)
|
| 41 |
predicted_class = torch.argmax(outputs.logits).item()
|
|
|
|
| 43 |
print("Predicted class:", "prevention" if predicted_class == 0 else "promotion")
|
| 44 |
```
|
| 45 |
|
|
|
|
|
|
|
| 46 |
## Labels
|
| 47 |
|
| 48 |
| Class | Description |
|
|
|
|
| 81 |
number={1},
|
| 82 |
year={2023}
|
| 83 |
}
|
| 84 |
+
```
|