Spaces:
Build error
Build error
Commit
·
4770cb5
1
Parent(s):
77a1b68
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("chatbotmodel.pth", map_location=torch.device('cpu'))
|
| 23 |
|
| 24 |
input_size = data["input_size"]
|
| 25 |
hidden_size = data["hidden_size"]
|