edwjin commited on
Commit
bb4e6bd
·
verified ·
1 Parent(s): a9d8847

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -43,7 +43,7 @@ tokenizer.vocab_size = len(tokenizer.stoi)
43
  model = Classifier(tokenizer.vocab_size)
44
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
45
  print('loading model')
46
- model.load_state_dict(torch.load('speechesdataset/classifier_model_dict.pth', map_location=device))
47
  print('finished loading model')
48
  model.to(device)
49
  model.eval()
 
43
  model = Classifier(tokenizer.vocab_size)
44
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
45
  print('loading model')
46
+ model.load_state_dict(torch.load('speechesdataset/checkpoint.pt', map_location=device))
47
  print('finished loading model')
48
  model.to(device)
49
  model.eval()