SHSH0819 commited on
Commit
c83b960
1 Parent(s): 7011da5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ def predict(data):
43
  model = DistillBERTClass('distilbert-base-cased')
44
 
45
  #print("================ model init ===========================")
46
- pretrained_model=torch.load(model_path + "/event_domain_final.pt",map_location=torch.device('cpu'))
47
  model.load_state_dict(pretrained_model['model_state_dict'])
48
  model.to(device)
49
 
 
43
  model = DistillBERTClass('distilbert-base-cased')
44
 
45
  #print("================ model init ===========================")
46
+ pretrained_model=torch.load(model_path + "event_domain_final.pt",map_location=torch.device('cpu'))
47
  model.load_state_dict(pretrained_model['model_state_dict'])
48
  model.to(device)
49