Hector Lopez commited on
Commit
1767191
1 Parent(s): c5b702e

Changed local files to False

Browse files
Files changed (1) hide show
  1. backend.py +1 -1
backend.py CHANGED
@@ -13,7 +13,7 @@ model = BertForSequenceClassification.from_pretrained(
13
  num_labels=5,
14
  output_attentions=False,
15
  output_hidden_states=False,
16
- local_files_only=True,
17
  )
18
  model.load_state_dict(torch.load("data/BERT_ft_epoch5.model"))
19
  model.eval()
 
13
  num_labels=5,
14
  output_attentions=False,
15
  output_hidden_states=False,
16
+ local_files_only=False,
17
  )
18
  model.load_state_dict(torch.load("data/BERT_ft_epoch5.model"))
19
  model.eval()