Update README.md

README.md

@@ -17,8 +17,8 @@ fake_sentence = "Za hory, za doly, kočka zlaté parohy"
 
 fake_sentence_tokens = ["[CLS]"] + tokenizer.tokenize(fake_sentence) + ["[SEP]"]
 fake_inputs = tokenizer.encode(fake_sentence, return_tensors="pt")
-discriminator_outputs = discriminator(fake_inputs)
-predictions = torch.nn.Sigmoid()(discriminator_outputs[0]).cpu().detach().numpy()
+outputs = discriminator(fake_inputs)
+predictions = torch.nn.Sigmoid()(outputs[0]).cpu().detach().numpy()
 
 for token in fake_sentence_tokens:
     print("{:>7s}".format(token), end="")