xjlulu commited on
Commit
849c4b7
1 Parent(s): 9106e6c
app.py CHANGED
@@ -46,7 +46,7 @@ best_model = SeqClassifier(
46
  ).to(device)
47
 
48
  # Define the path to the checkpoint file
49
- ckpt_path = ckpt_dir / "model_checkpoint.pth"
50
 
51
  # Load the model's weights
52
  checkpoint = torch.load(ckpt_path, map_location=torch.device('cpu'))
 
46
  ).to(device)
47
 
48
  # Define the path to the checkpoint file
49
+ ckpt_path = ckpt_dir / "intent_checkpoint.pth"
50
 
51
  # Load the model's weights
52
  checkpoint = torch.load(ckpt_path, map_location=torch.device('cpu'))
ckpt/intent/{model_checkpoint.pth → intent_checkpoint.pth} RENAMED
File without changes