Update app.py

app.py

@@ -275,6 +275,18 @@ def safe_remove_dir(directory):
             print(f"Unexpected error while deleting {directory}: {e}")
 
 
+def pad_to_30_seconds(waveform, sample_rate):
+    target_len = 30 * sample_rate
+    current_len = waveform.shape[-1]
+
+    if current_len >= target_len:
+        return waveform[:, :target_len]  # Truncate if longer
+    else:
+        pad_len = target_len - current_len
+        padding = torch.zeros((waveform.shape[0], pad_len))
+        padded_waveform = torch.cat((waveform, padding), dim=1)
+        return padded_waveform
+
 class Music2emo:
     def __init__(
         self,
@@ -377,12 +389,13 @@ class Music2emo:
         mert_dir.mkdir(parents=True, exist_ok=True)
 
         waveform, sample_rate = torchaudio.load(audio)
+        waveform = pad_to_30_seconds(waveform, sample_rate)
+        
         if waveform.shape[0] > 1:
             waveform = waveform.mean(dim=0).unsqueeze(0)
         waveform = waveform.squeeze()
         waveform, sample_rate = resample_waveform(waveform, sample_rate, resample_rate)
 
-
         # 🔍 Check duration
         duration_sec = waveform.shape[-1] / sample_rate
         is_split = duration_sec > 30.0