Update model.py

app/model.py

@@ -66,7 +66,7 @@ path_audio_model_2 = load_model(config_data.model_audio_url, config_data.model_a
 audio_processor = AutoFeatureExtractor.from_pretrained(path_audio_model_1)
 
 audio_model = ExprModelV3.from_pretrained(path_audio_model_1)
-audio_model.load_state_dict(torch.load(path_audio_model_2)["model_state_dict"])
+audio_model.load_state_dict(torch.load(path_audio_model_2, map_location=torch.device(device))["model_state_dict"])
 audio_model.to(device)
 audio_model.eval()
 audio_model(test_audio.to(device))