DHRUV SHEKHAWAT
commited on
Commit
·
2eacd00
1
Parent(s):
f4fcfa8
Update app.py
Browse files
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 |
|