Updated API code

README.md

@@ -74,13 +74,11 @@ data = {"inputs": prompt, "parameters": {"top_k": 3}}
 
 response = requests.request("POST", api_url, headers=headers, json=data)
 
-# The labels are arranged according to likelihood of classification
-repdict = {"LABEL_0": "Pow", "LABEL_1": "Ach", "LABEL_2": "Aff"}
-# so we replace them in the output
-scores = {repdict[x['label']]: x['score'] for x in response.json()}
+# Print the labels and scores (arranged in order of likelihood)
+scores = {x['label']: x['score'] for x in response.json()}
 print(scores)
 
-# output: {'Pow': 0.8279141187667847, 'Aff': 0.7250356674194336, 'Ach': 0.0020263446494936943}
+# {'Aff': 0.999998927116394, 'Pow': 0.999890923500061, 'Ach': 5.351924119167961e-05}
 ```
 
 ## References