parkermoe commited on
Commit
f671f31
·
1 Parent(s): ab19f0a

'update model path cpu'

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = 'checkpoint_epoch_99.pth'
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')