Update pages/19_RNN_LSTM_Shakespeare.py

pages/19_RNN_LSTM_Shakespeare.py

@@ -22,7 +22,7 @@ class LSTMModel(nn.Module):
 def generate_text(model, start_str, length, char_to_int, int_to_char, num_layers, hidden_size):
     model.eval()
     input_seq = [char_to_int[c] for c in start_str]
-    input_seq = torch.tensor(input_seq, dtype=torch.float32).unsqueeze(0).unsqueeze(-1)
+    input_seq = torch.tensor(input_seq, dtype=torch.float32).unsqueeze(0).unsqueeze(-1)  # Shape: (1, seq_length, 1)
     h = (torch.zeros(num_layers, 1, hidden_size), torch.zeros(num_layers, 1, hidden_size))
     generated_text = start_str
 
@@ -110,4 +110,4 @@ if st.button("Train and Generate"):
             # Text generation
             generated_text = generate_text(model, start_string, generate_length, char_to_int, int_to_char, num_layers, hidden_size)
             st.subheader("Generated Text")
-            st.write(generated_text)
+            st.write(generated_text)