Update app.py

app.py

@@ -13,15 +13,11 @@ from model.lstm_model import BiLSTM
 # Загрузка весов модели (выполняется только при первом запуске)
 @st.cache_data
 def load_model_weights():
-    model = BiLSTM(input_size, hidden_size, num_layers, output_size)
-    model.load_state_dict(torch.load('model/model_weights.pth'))
-    model.eval()
-    return model
-model_weights = load_model_weights()
+    return torch.load('model/model_weights.pth')
 
 # Сохранение весов модели в сессионном состоянии
 if 'model_weights' not in st.session_state:
-    st.session_state.model_weights = model_weights
+    st.session_state.model_weights = load_model_weights()
 
 st.set_page_config(layout='wide', initial_sidebar_state='expanded')
 st.set_option('deprecation.showPyplotGlobalUse', False)