Update pipeline.py

pipeline.py

@@ -1,3 +1,7 @@
+import torch
+import torch.nn as nn
+import random
+from huggingface_hub import PyTorchModelHubMixin
 
 class LSTMTextGenerator(nn.Module, PyTorchModelHubMixin):
     def __init__(self, input_size=45, hidden_size=512, output_size=45, num_layers=2, dropout=0.5):
@@ -23,7 +27,7 @@ class LSTMTextGenerator(nn.Module, PyTorchModelHubMixin):
 class PreTrainedPipeline():
     def __init__(self, path=""):
         self.model = LSTMTextGenerator.from_pretrained("miittnnss/lstm-textgen-pets")
-        self.chars = ""
+        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)