Update pages/3_BiLTSM_Прогноз bitcoin.py

pages/3_BiLTSM_Прогноз bitcoin.py

@@ -60,10 +60,11 @@ model = BiLSTM(input_size, hidden_size, num_layers, output_size)
 criterion = nn.MSELoss()
 optimizer = optim.Adam(model.parameters(), lr=0.001)
 
+# Получение весов модели из сессионного состояния
 model_weights = st.session_state.model_weights
 
-# Использование весов модели в дальнейших вычислениях
-model.load_state_dict(model_weights.state_dict())
+# Загрузка весов в модель
+model.load_state_dict(model_weights)
 model.eval()
 with torch.no_grad():
     test_predictions = model(X_test)