Update README.md

README.md

@@ -42,11 +42,17 @@ Or Python API:
 ```
 from transformers import AutoModelForSequenceClassification, AutoTokenizer
 
+import numpy as np
+
+labdic ={ 0: "rationale", 1: "suggestion", 2: "specific_critique"}
+
 model = AutoModelForSequenceClassification.from_pretrained("Shenzy/Sentence_Classification4DesignTutor", use_auth_token=True)
 
 tokenizer = AutoTokenizer.from_pretrained("Shenzy/Sentence_Classification4DesignTutor", use_auth_token=True)
 
-inputs = tokenizer("I love AutoTrain", return_tensors="pt")
+inputs = tokenizer("An unusual hierarchy in the section near the top where the design seems to prioritise running time over a compacted artist name.", return_tensors="pt")
 
 outputs = model(**inputs)
+
+print(labdic[np.argmax(outputs)])
 ```