Spaces:
Build error
Build error
ShAnSantosh
commited on
Commit
•
1323c4c
1
Parent(s):
cbf9b8e
Update app.py
Browse files
app.py
CHANGED
@@ -19,7 +19,7 @@ with open('./intents.json', 'r') as json_data:
|
|
19 |
|
20 |
FILE = "./data.pth"
|
21 |
data = torch.load(FILE)
|
22 |
-
model_state = torch.load("
|
23 |
|
24 |
input_size = data["input_size"]
|
25 |
hidden_size = data["hidden_size"]
|
|
|
19 |
|
20 |
FILE = "./data.pth"
|
21 |
data = torch.load(FILE)
|
22 |
+
model_state = torch.load("chatmodel.pth", map_location=torch.device('cpu'))
|
23 |
|
24 |
input_size = data["input_size"]
|
25 |
hidden_size = data["hidden_size"]
|