update demo.

shibing624 · Apr 6, 2021 · af26796 · af26796
1 parent cbd6dcc
commit af26796
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/pycorrector/electra/predict_mask.py b/pycorrector/electra/predict_mask.py
@@ -40,7 +40,6 @@ def detect_error_demo():
     fake_tokens = tokenizer.tokenize(sentence)
     print(fake_tokens)
     fake_inputs = tokenizer.encode(sentence, return_tensors="pt")
-
     discriminator_outputs = discriminator(fake_inputs)
     predictions = torch.round((torch.sign(discriminator_outputs[0]) + 1) / 2)