Nicolas Denier commited on
Commit
baff0a5
·
1 Parent(s): fbc8bd3

correct file path

Browse files
Files changed (1) hide show
  1. tasks/audio.py +1 -1
tasks/audio.py CHANGED
@@ -46,7 +46,7 @@ async def evaluate_audio(request: AudioEvaluationRequest):
46
  # Load model
47
  model = ChainsawDetector(batch_size).to(device, dtype=torch.bfloat16)
48
  model = torch.compile(model)
49
- model.load_state_dict(torch.load('models/final-bf16.pth', weights_only=True))
50
  model.eval()
51
 
52
  # Start tracking emissions
 
46
  # Load model
47
  model = ChainsawDetector(batch_size).to(device, dtype=torch.bfloat16)
48
  model = torch.compile(model)
49
+ model.load_state_dict(torch.load('tasks/models/final-bf16.pth', weights_only=True))
50
  model.eval()
51
 
52
  # Start tracking emissions