Update model.py

model.py

@@ -5,7 +5,7 @@ class RNN_model(nn.Module):
   def __init__(self):
     super().__init__()
 
-    self.rnn= nn.RNN(input_size=1080, hidden_size=240,num_layers=1, nonlinearity= 'relu', bias= True)
+    self.rnn= nn.GRU(input_size=1080, hidden_size=240,num_layers=1, nonlinearity= 'relu', bias= True)
     self.output= nn.Linear(in_features=240, out_features=24)
 
   def forward(self, x):