Nicolas Denier
commited on
Commit
·
baff0a5
1
Parent(s):
fbc8bd3
correct file path
Browse files- 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
|