Update app.py

app.py

@@ -202,27 +202,43 @@ def resample_waveform(waveform, original_sample_rate, target_sample_rate):
 #         segments.append(waveform)
 
 #     return segments
-
-
-def split_audio(waveform, sample_rate, segment_duration=10):
+def split_audio(waveform, sample_rate):
     segment_samples = segment_duration * sample_rate
     total_samples = waveform.size(0)
 
+    # Pad if shorter than one segment
+    if total_samples < segment_samples:
+        pad_size = segment_samples - total_samples
+        waveform = torch.nn.functional.pad(waveform, (0, pad_size))
+
     segments = []
-    for start in range(0, total_samples, segment_samples):
+    for start in range(0, waveform.size(0), segment_samples):
         end = start + segment_samples
-        if end <= total_samples:
+        if end <= waveform.size(0):
             segment = waveform[start:end]
             segments.append(segment)
 
-    # If no full segments were created, pad the short waveform
-    if len(segments) == 0:
-        pad_length = segment_samples - total_samples
-        padded_waveform = torch.nn.functional.pad(waveform, (0, pad_length))
-        segments.append(padded_waveform)
-
     return segments
 
+# def split_audio(waveform, sample_rate, segment_duration=10):
+#     segment_samples = segment_duration * sample_rate
+#     total_samples = waveform.size(0)
+
+#     segments = []
+#     for start in range(0, total_samples, segment_samples):
+#         end = start + segment_samples
+#         if end <= total_samples:
+#             segment = waveform[start:end]
+#             segments.append(segment)
+
+#     # If no full segments were created, pad the short waveform
+#     if len(segments) == 0:
+#         pad_length = segment_samples - total_samples
+#         padded_waveform = torch.nn.functional.pad(waveform, (0, pad_length))
+#         segments.append(padded_waveform)
+
+#     return segments
+
 # def split_audio(waveform, sample_rate):
 #     segment_samples = segment_duration * sample_rate
 #     total_samples = waveform.size(0)