Update README.md

README.md

@@ -1,3 +1,43 @@
----
-license: apache-2.0
----
+---
+license: apache-2.0
+---
+
+```python
+from transformers import AutoTokenizer, AutoModelForSequenceClassification
+import numpy as np
+
+def sigmoid(x):
+    return 1/(1 + np.exp(-x))
+
+classes = [
+ 'code_design',
+ 'code_style',
+ 'code_naming',
+ 'code_logic',
+ 'code_io',
+ 'code_data',
+ 'code_doc',
+ 'code_api',
+ 'compatibility',
+ 'rule_def',
+ 'config_commit_patch_review',
+ 'config_building_installing',
+]
+class2id = {class_:id for id, class_ in enumerate(classes)}
+id2class = {id:class_ for class_, id in class2id.items()}
+
+checkpoint = 'mochodek/bert4comment-subject'
+tokenizer = AutoTokenizer.from_pretrained(checkpoint)
+model = AutoModelForSequenceClassification.from_pretrained(checkpoint)
+
+text = "What do you think about making this constant?"
+encoded_input = tokenizer(text, return_tensors='pt')
+output = model(**encoded_input)
+
+logits = output.logits.detach().numpy()
+
+scores = sigmoid(logits)
+scores = (scores > 0.5).astype(int).reshape(-1)
+scores_labels = [class_name for class_name in classes if scores[class2id[class_name]] == 1 ]
+
+```