Update README.md

README.md

@@ -1,3 +1,31 @@
----
-license: apache-2.0
----
+---
+license: apache-2.0
+---
+
+```python
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+from scipy.special import softmax
+
+checkpoint = 'mochodek/bert4code-purpose'
+tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+
+model = AutoModelForSequenceClassification.from_pretrained(checkpoint)
+
+id2class = {
+    0: 'discussion_participation',
+    1: 'discussion_trigger',
+    2: 'change_request',
+    3: 'acknowledgement',
+    4: 'same_as'
+}
+
+text = "Please, make constant from that string"
+encoded_input = tokenizer(text, return_tensors='pt')
+
+output = model(**encoded_input)
+
+scores = softmax(output.logits.cpu().detach().numpy())
+
+id2class[np.argmax(scores)]
+
+```