typo fix for get_probabilities

README.md

@@ -79,7 +79,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()
@@ -92,7 +92,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: