removed state_dict

app.py

@@ -80,7 +80,7 @@ def simulate_risk_score(rainfall, evaporation, inflow, outflow, population, wate
 input_size = 14
 output_size = 3
 model = EnhancedLSTM(input_size=input_size, lstm_layer_sizes=[128]*3, linear_layer_size=[64]*6, output_size=output_size)
-model.load_state_dict(torch.load('model/res_1.pt', map_location='cpu', weights_only=False))
+model.load_state_dict(torch.load('./model/res_1.pt', map_location='cpu'))
 model.eval()
 
 

src/model.py

@@ -48,4 +48,6 @@ class EnhancedLSTM(nn.Module):
                 if 'weight' in name:
                     nn.init.xavier_uniform_(param.data)
                 elif 'bias' in name:
-                    nn.init.zeros_(param.data)
+                    nn.init.zeros_(param.data)
+
+