Rename pages/19_RNN_Shakespeare.py to pages/19_RNN_LSTM_Shakespeare.py

pages/{19_RNN_Shakespeare.py → 19_RNN_LSTM_Shakespeare.py}

@@ -94,8 +94,8 @@ if st.button("Train and Generate"):
                     targets = Y_tensor[i].unsqueeze(0)
 
                     # Forward pass
-                    outputs, h = model(inputs, h)
-                    h = (h[0].detach(), h[1].detach())
+                    outputs, h = model(inputs, (h[0].squeeze(1), h[1].squeeze(1)))
+                    h = (h[0].detach().unsqueeze(1), h[1].detach().unsqueeze(1))
                     loss = criterion(outputs, targets)
 
                     # Backward pass and optimization