Update app.py

app.py

@@ -24,6 +24,8 @@ def sliding_windows(data, seq_length):
 class LSTM(nn.Module):
     def __init__(self, input_size, hidden_size, num_layers, output_size):
         super(LSTM, self).__init__()
+        self.hidden_size = hidden_size
+        self.num_layers = num_layers
         self.lstm = nn.LSTM(input_size, hidden_size, num_layers, batch_first=True)
         self.fc = nn.Linear(hidden_size, output_size)
 
@@ -37,6 +39,8 @@ class LSTM(nn.Module):
 class GRU(nn.Module):
     def __init__(self, input_size, hidden_size, num_layers, output_size):
         super(GRU, self).__init__()
+        self.hidden_size = hidden_size
+        self.num_layers = num_layers
         self.gru = nn.GRU(input_size, hidden_size, num_layers, batch_first=True)
         self.fc = nn.Linear(hidden_size, output_size)