update app.py

app.py

@@ -69,7 +69,9 @@ class CNN1DLSTMAudioClassifier(nn.Module):
 num_class = 6
 model = CNN1DLSTMAudioClassifier(num_class)
 
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.load_state_dict(torch.load("speech-emotion-recognition-best-model.bin", weights_only=False))
+model = model.to(device)
 model.eval()
 
 def preprocess_single_audio(file_path, sample_rate=16000, n_mels=128, n_fft=2048, hop_length=512):
@@ -135,8 +137,6 @@ def decode_emotion_prediction(prediction_tensor, label_encoder):
     
     return predicted_emotion, confidence
 
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = model.to(device)
 
 def predict(wave):
     wave = preprocess_single_audio(wave)