Update app.py

app.py

@@ -19,7 +19,7 @@ class MultiTaskBertModel(nn.Module):
 
 # Загрузка сохраненной модели
 model = MultiTaskBertModel(num_labels_task1=3, num_labels_task2=4)
-model.load_state_dict(torch.load("ticket_classifier.pth"))
+model.load_state_dict(torch.load("ticket_classifier.pth", map_location=torch.device('cpu')))
 model.eval()
 
 # Загрузка токенизатора