Update app.py

app.py

@@ -68,9 +68,8 @@ class CNN1DLSTMAudioClassifier(nn.Module):
     
 num_class = 6
 model = CNN1DLSTMAudioClassifier(num_class)
-device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = model.load_state_dict(torch.load("best-model-emotion-recognition.bin", weights_only=False))
-model = model.to(device)
+
+model.load_state_dict(torch.load("best-model-emotion-recognition.bin", weights_only=False))
 model.eval()
 
 def preprocess_single_audio(file_path, sample_rate=16000, n_mels=128, n_fft=2048, hop_length=512):
@@ -136,13 +135,14 @@ 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)
     le = LabelEncoder()
     le.classes_ = np.array(['Angry', 'Disgusting', 'Fear', 'Happy', 'Neutral', 'Sad'])
     wave = wave.unsqueeze(0)
-    wave = wave.to(device)
     with torch.no_grad():
         prediction = model(wave)
     predicted_emotion, confidence = decode_emotion_prediction(prediction, le)