Fixed small summary with model architectures until now

src/models/lstm_model.py

@@ -37,9 +37,9 @@ class GRUModel(torch.nn.Module):
 
     def forward(self, x):
         # Forward pass through the GRU layers
-        _, hidden_state = self.gru(x)
-        
+        x, _ = self.gru(x)
+        x = x[:, -1, :]
         # Use the hidden state from the last time step for the output
-        output = self.linear(hidden_state[-1])
+        output = self.linear(x)
         
         return output