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'))
|