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

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

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