Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -43,7 +43,7 @@ class MultiTaskBertModel(torch.nn.Module):
|
|
43 |
model = cls(bert_model, num_labels_task1, num_labels_task2)
|
44 |
|
45 |
# Загружаем сохраненные веса
|
46 |
-
model_path = os.path.join('
|
47 |
model.load_state_dict(torch.load(model_path))
|
48 |
|
49 |
return model
|
|
|
43 |
model = cls(bert_model, num_labels_task1, num_labels_task2)
|
44 |
|
45 |
# Загружаем сохраненные веса
|
46 |
+
model_path = os.path.join('pytorch_model.bin')
|
47 |
model.load_state_dict(torch.load(model_path))
|
48 |
|
49 |
return model
|