Spaces:
Sleeping
Sleeping
parkermoe
commited on
Commit
·
7968e20
1
Parent(s):
f9e4c0f
'fixed path'
Browse files
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 = '/
|
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}")
|