DHRUV SHEKHAWAT commited on
Commit
2eacd00
·
1 Parent(s): f4fcfa8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -52,7 +52,7 @@ def evaluate(transformer, question, question_mask, max_len, word_map):
52
 
53
 
54
  if load_checkpoint:
55
- checkpoint = torch.load(ckpt_path)
56
  transformer = checkpoint['transformer']
57
 
58
 
 
52
 
53
 
54
  if load_checkpoint:
55
+ checkpoint = torch.load(ckpt_path, map_location=torch.device('cpu'))
56
  transformer = checkpoint['transformer']
57
 
58