Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -11,7 +11,7 @@ from datetime import date
|
|
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'))
|
|
|
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'))
|