Fix typo in example code (#3)

- Fix typo in example code (c5c885a8066c37e68cce6bb34823442b0488ac09)


Co-authored-by: Jeong Deok <[email protected]>

README.md

@@ -80,7 +80,7 @@ def parse_output(output, input_len):
         list_index_logprobs_i = [torch.topk(token_i, k=nlogprobs, largest=True, sorted=True)
                                  for token_i in list(output.scores)[:-1]]
         if list_index_logprobs_i is not None:
-            prob = get_probablities(list_index_logprobs_i)
+            prob = get_probabilities(list_index_logprobs_i)
             prob_of_risk = prob[1]
 
     res = tokenizer.decode(output.sequences[:,input_len:][0],skip_special_tokens=True).strip()
@@ -93,7 +93,7 @@ def parse_output(output, input_len):
 
     return label, prob_of_risk.item()
 
-def get_probablities(logprobs):
+def get_probabilities(logprobs):
     safe_token_prob = 1e-50
     unsafe_token_prob = 1e-50
     for gen_token_i in logprobs: