Update app.py

app.py

@@ -14,10 +14,19 @@ try:
 except ImportError as e:
     print(f"AudioSeal could not be imported: {e}")
 
-def extract_mfcc_features(waveform, sample_rate, n_mfcc=13):
-    mfcc_transform = T.MFCC(sample_rate=sample_rate, n_mfcc=n_mfcc)
+def extract_mfcc_features(waveform, sample_rate, n_mfcc=40, n_mels=128, win_length=400, hop_length=160):
+    mfcc_transform = T.MFCC(
+        sample_rate=sample_rate,
+        n_mfcc=n_mfcc,
+        melkwargs={
+            'n_fft': 400,
+            'n_mels': n_mels,
+            'hop_length': hop_length,
+            'win_length': win_length
+        }
+    )
     mfcc = mfcc_transform(waveform)
-    return mfcc.mean(dim=2)
+    return mfcc.mean(dim=2) 
 
 def plot_spectrogram(waveform, sample_rate):
     if waveform.ndim == 1: