Update app.py

app.py

@@ -126,21 +126,14 @@ class GRUModel(nn.Module):
         return self.fc(out[:, -1, :])
 
 class BiLSTMModel(nn.Module):
-    def __init__(self, input_dim=1, hidden_dim=100, num_layers=2, output_dim=1, dropout_prob=0.2):
+    def __init__(self):
         super(BiLSTMModel, self).__init__()
-        self.lstm = nn.LSTM(
-            input_size=input_dim,
-            hidden_size=hidden_dim,
-            num_layers=num_layers,
-            batch_first=True,
-            dropout=dropout_prob,
-            bidirectional=True
-        )
-        self.fc = nn.Linear(hidden_dim * 2, output_dim)  # because bidirectional
+        self.lstm = nn.LSTM(input_size=1, hidden_size=100, num_layers=2, batch_first=True, dropout=0.2, bidirectional=True)
+        self.fc = nn.Linear(200, 1)
 
     def forward(self, x):
-        h0 = torch.zeros(self.lstm.num_layers * 2, x.size(0), self.lstm.hidden_size).to(x.device)
-        c0 = torch.zeros(self.lstm.num_layers * 2, x.size(0), self.lstm.hidden_size).to(x.device)
+        h0 = torch.zeros(4, x.size(0), 100)  # 2 directions × 2 layers = 4
+        c0 = torch.zeros(4, x.size(0), 100)
         out, _ = self.lstm(x, (h0, c0))
         return self.fc(out[:, -1, :])
 @st.cache_resource(ttl=3600)