jjamnicki commited on
Commit
7990cde
1 Parent(s): 6ca0f97

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -82,7 +82,7 @@ texts = [
82
  ]
83
 
84
  # prepare the input
85
- encoding = formality_tokenizer(
86
  texts,
87
  add_special_tokens=True,
88
  return_token_type_ids=True,
@@ -92,7 +92,7 @@ encoding = formality_tokenizer(
92
  )
93
 
94
  # inference
95
- output = formality_model(**formality_encoding)
96
 
97
  formality_scores = [
98
  {id2formality[idx]: score for idx, score in enumerate(text_scores.tolist())}
 
82
  ]
83
 
84
  # prepare the input
85
+ encoding = tokenizer(
86
  texts,
87
  add_special_tokens=True,
88
  return_token_type_ids=True,
 
92
  )
93
 
94
  # inference
95
+ output = model(**encoding)
96
 
97
  formality_scores = [
98
  {id2formality[idx]: score for idx, score in enumerate(text_scores.tolist())}