Update tasks/audio.py

tasks/audio.py

@@ -94,7 +94,7 @@ async def evaluate_audio(request: AudioEvaluationRequest):
     model = BlazeFaceModel(input_channels=1, label_count=model_settings['label_count'], use_double_block=False, activation='relu', use_optional_block=False)
 
 
-    model.load_state_dict(torch.load("./best_blazeface_model.pth", map_location=torch.device('cpu')))
+    model.load_state_dict(torch.load("./best_blazeface_model.pth", map_location=torch.device('cpu'), weights_only=True))
 
     model = torch.quantization.quantize_dynamic(model, {torch.nn.Linear}, dtype=torch.qint8)