'fix <pad>'

app.py

@@ -78,7 +78,8 @@ class TextGenerator:
                 next_token = self.sample_from(next_token_logits, temperature)
                 tokens = torch.cat([tokens, torch.LongTensor([[next_token]]).to(device)], dim=1)
 
-        generated_text = ' '.join(self.vocab.get_itos()[token] for token in tokens[0])
+        generated_tokens = [token for token in tokens[0] if self.vocab.get_itos()[token] != '<pad>']
+        generated_text = ' '.join(self.vocab.get_itos()[token] for token in generated_tokens)
         return generated_text
     
 text_generator = TextGenerator(vocab=vocab, top_k=10)