Dzhamb commited on
Commit
f8a5603
1 Parent(s): 1b85e88

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +1 -1
utils.py CHANGED
@@ -33,7 +33,7 @@ def get_labels(text, model, tokenizer, count_labels=8):
33
  result_labels.append(pair[1])
34
 
35
 
36
- def load_model():
37
  tokenizer = DistilBertTokenizerFast()
38
  model = DistilBertForSequenceClassification()
39
  model.load_state_dict(torch.load('weight_model'))
 
33
  result_labels.append(pair[1])
34
 
35
 
36
+ def load_model():
37
  tokenizer = DistilBertTokenizerFast()
38
  model = DistilBertForSequenceClassification()
39
  model.load_state_dict(torch.load('weight_model'))