Spaces:
Sleeping
Sleeping
parkermoe commited on
Commit ·
f671f31
1
Parent(s): ab19f0a
'update model path cpu'
Browse files
app.py
CHANGED
|
@@ -37,7 +37,7 @@ class LSTMModel(nn.Module):
|
|
| 37 |
model = LSTMModel(VOCAB_SIZE, EMBEDDING_DIM, N_UNITS).to(device)
|
| 38 |
|
| 39 |
|
| 40 |
-
checkpoint_path = '
|
| 41 |
checkpoint = torch.load(checkpoint_path)
|
| 42 |
model.load_state_dict(checkpoint['model_state_dict'])
|
| 43 |
print('Loaded model from checkpoint')
|
|
|
|
| 37 |
model = LSTMModel(VOCAB_SIZE, EMBEDDING_DIM, N_UNITS).to(device)
|
| 38 |
|
| 39 |
|
| 40 |
+
checkpoint_path = 'recipe_generator_LSTM.pth'
|
| 41 |
checkpoint = torch.load(checkpoint_path)
|
| 42 |
model.load_state_dict(checkpoint['model_state_dict'])
|
| 43 |
print('Loaded model from checkpoint')
|