niknikita commited on
Commit
af9102e
·
1 Parent(s): f342497

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -34,7 +34,7 @@ def get_top95(y_predict, convert_target):
34
  return lst_labels
35
  #
36
  # model = MyModel()
37
- model = torch.load("model.pt", map_location='cpu').eval()
38
 
39
  # print(model)
40
  # model = DistilBertForSequenceClassification.from_pretrained("model/distilbert-model1.pt", local_files_only=True)
 
34
  return lst_labels
35
  #
36
  # model = MyModel()
37
+ model = torch.load("model.pt", map_location='cpu')
38
 
39
  # print(model)
40
  # model = DistilBertForSequenceClassification.from_pretrained("model/distilbert-model1.pt", local_files_only=True)