parkermoe commited on
Commit
7968e20
·
1 Parent(s): f9e4c0f

'fixed path'

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -36,13 +36,13 @@ class LSTMModel(nn.Module):
36
  model = LSTMModel(VOCAB_SIZE, EMBEDDING_DIM, N_UNITS).to(device)
37
 
38
 
39
- checkpoint_path = '/Users/parkermoesta/Library/Mobile Documents/com~apple~CloudDocs/Generative Models/LSTM/recipe_generator_LSTM/checkpoint_epoch_99.pth'
40
  checkpoint = torch.load(checkpoint_path)
41
  model.load_state_dict(checkpoint['model_state_dict'])
42
  print('Loaded model from checkpoint')
43
 
44
  def load_vocab(directory):
45
- file_path = os.path.join(directory, 'vocab.pkl')
46
  with open(file_path, 'rb') as input:
47
  vocab = pickle.load(input)
48
  print(f"Vocabulary loaded from {file_path}")
 
36
  model = LSTMModel(VOCAB_SIZE, EMBEDDING_DIM, N_UNITS).to(device)
37
 
38
 
39
+ checkpoint_path = 'RecipeGenerator/checkpoint_epoch_99.pth'
40
  checkpoint = torch.load(checkpoint_path)
41
  model.load_state_dict(checkpoint['model_state_dict'])
42
  print('Loaded model from checkpoint')
43
 
44
  def load_vocab(directory):
45
+ file_path = os.path.join(directory, 'RecipeGenerator/vocab.pkl')
46
  with open(file_path, 'rb') as input:
47
  vocab = pickle.load(input)
48
  print(f"Vocabulary loaded from {file_path}")