Bagfix

src/cyberclaasic.py

@@ -39,7 +39,8 @@ class CyberClassic(torch.nn.Module):
 
         decoded = self.tokenizer.batch_decode(generated, skip_special_tokens=True)
 
-        score = self.discriminator(decoded)
+        decoded_tokens = self.discriminator_tokenizer(decoded, return_tensors='pt', padding=True, truncation=True)
+        score = self.discriminator(decoded_tokens)
         index = int(torch.argmax(score))
 
         return decoded[index]