DmitriySv commited on
Commit
6b9c12e
·
verified ·
1 Parent(s): c55ebf8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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('https://huggingface.co', load_directory, 'resolve/main', 'pytorch_model.bin')
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