saintyboy commited on
Commit
b6e3bcc
1 Parent(s): 352557d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -27,7 +27,7 @@ enc = tiktoken.Encoding(
27
  )
28
 
29
  # Load model from checkpoint
30
- model_save_path = 'latest_model.pt'
31
  if os.path.exists(model_save_path):
32
  model = torch.load(model_save_path, map_location=device)
33
  else:
 
27
  )
28
 
29
  # Load model from checkpoint
30
+ model_save_path = 'tuned_ckpt.pt'
31
  if os.path.exists(model_save_path):
32
  model = torch.load(model_save_path, map_location=device)
33
  else: