Spaces:
Build error
Build error
Commit
·
05bc6a7
1
Parent(s):
ecdd66c
Update app.py
Browse files
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"]
|