Update pipeline.py

pipeline.py

@@ -30,7 +30,7 @@ class PreTrainedPipeline():
         self.chars = "!',.;ACDFGHIMORSTWabcdefghijklmnopqrstuvwxy"
         self.char_to_index = {char: index for index, char in enumerate(self.chars)}
         self.index_to_char = {index: char for char, index in self.char_to_index.items()}
-        self.output_size = len(chars)
+        self.output_size = len(self.chars)
         self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
     def __call__(self, inputs: str):