DmitriySv commited on
Commit
5656ec0
·
verified ·
1 Parent(s): e4d7b62

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -19,7 +19,7 @@ class MultiTaskBertModel(nn.Module):
19
 
20
  # Загрузка сохраненной модели
21
  model = MultiTaskBertModel(num_labels_task1=3, num_labels_task2=4)
22
- model.load_state_dict(torch.load("ticket_classifier.pth", map_location=torch.device('cpu')))
23
  model.eval()
24
 
25
  # Загрузка токенизатора
 
19
 
20
  # Загрузка сохраненной модели
21
  model = MultiTaskBertModel(num_labels_task1=3, num_labels_task2=4)
22
+ model = torch.load('ticket_classifier.pth', weights_only=False, map_location=torch.device('cpu'))
23
  model.eval()
24
 
25
  # Загрузка токенизатора