parkermoe commited on
Commit
29d0dff
·
1 Parent(s): 13d544c

'update vocab path'

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -44,8 +44,8 @@ model.load_state_dict(checkpoint)
44
 
45
  print('Loaded model from checkpoint')
46
 
47
- def load_vocab(directory):
48
- file_path = os.path.join(directory, 'vocab.pkl')
49
  with open(file_path, 'rb') as input:
50
  vocab = pickle.load(input)
51
  print(f"Vocabulary loaded from {file_path}")
 
44
 
45
  print('Loaded model from checkpoint')
46
 
47
+ def load_vocab(file_path):
48
+ file_path = os.path.join(file_path)
49
  with open(file_path, 'rb') as input:
50
  vocab = pickle.load(input)
51
  print(f"Vocabulary loaded from {file_path}")