Update README.md

add usage example

README.md

@@ -27,6 +27,46 @@ The table connects the position of the predicted probability to its dimension.
 | 2         | Left-Wing Host-Ideology  |
 | 3         | Right-Wing Host-Ideology |
 
+# Usage Example
+
+```python
+import torch
+from transformers import AutoModel
+from transformers import AutoTokenizer
+
+# optional commit_hash to ensure a consistent version of the model
+commit_hash = "2354335caedc36df44da926291786f0159a502f0"
+
+# load tokenizer
+tokenizer = AutoTokenizer.from_pretrained("luerhard/PopBERT", revision=commit_hash)
+
+# load model
+# trust_remote_code is necessary to use the custom architecture of this model (module.py)
+model = AutoModel.from_pretrained("luerhard/PopBERT", trust_remote_code=True, revision=commit_hash)
+
+# define text to be predicted
+text = (
+    "Das ist Klassenkampf von oben, das ist Klassenkampf im Interesse von "
+    "Vermögenden und Besitzenden gegen die Mehrheit der Steuerzahlerinnen und "
+    "Steuerzahler auf dieser Erde."
+)
+
+# encode text with tokenizer
+encodings = tokenizer(text, padding=True, return_tensors="pt")
+
+# predict
+with torch.inference_mode():
+    _, prediction_tensor = model(**encodings)
+
+# convert prediction from torch tensor to numpy array
+prediction = prediction_tensor.numpy()
+print(prediction)
+```
+
+```
+[[0.84803474 0.9991047  0.9919584  0.19843338]]
+```
+
 
 # Performance