typo fix for get_probabilities (#4)

- typo fix for get_probabilities (e705b8a7651724b33bd4eb227a6ff4697aeaba82)


Co-authored-by: Manish Nagireddy <mnagired@users.noreply.huggingface.co>

README.md

@@ -76,7 +76,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()
@@ -89,7 +89,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: