Spaces:
Running
Running
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')
|