Solar-Iz commited on
Commit
c5ebcfd
·
verified ·
1 Parent(s): b5b1417

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ from datetime import date
11
  from model.lstm_model import BiLSTM
12
 
13
  # Загрузка весов модели (выполняется только при первом запуске)
14
- @st.cache_data(allow_output_mutation=True)
15
  def load_model_weights():
16
  model = BiLSTM(input_size, hidden_size, num_layers, output_size)
17
  model.load_state_dict(torch.load('model/model_weights.pth'))
 
11
  from model.lstm_model import BiLSTM
12
 
13
  # Загрузка весов модели (выполняется только при первом запуске)
14
+ @st.cache_data
15
  def load_model_weights():
16
  model = BiLSTM(input_size, hidden_size, num_layers, output_size)
17
  model.load_state_dict(torch.load('model/model_weights.pth'))