ShAnSantosh commited on
Commit
05bc6a7
·
1 Parent(s): ecdd66c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -18,7 +18,7 @@ with open('./intents.json', 'r') as json_data:
18
  intents = json.load(json_data)
19
 
20
  FILE = "./data.pth"
21
- data = torch.load(FILE)
22
 
23
  input_size = data["input_size"]
24
  hidden_size = data["hidden_size"]
 
18
  intents = json.load(json_data)
19
 
20
  FILE = "./data.pth"
21
+ data = torch.load(FILE, map_location=torch.device('cpu'))
22
 
23
  input_size = data["input_size"]
24
  hidden_size = data["hidden_size"]