Update app.py

app.py

@@ -203,7 +203,8 @@ def resample_waveform(waveform, original_sample_rate, target_sample_rate):
 
 #     return segments
 
-def split_audio(waveform, sample_rate):
+
+def split_audio(waveform, sample_rate, segment_duration=10):
     segment_samples = segment_duration * sample_rate
     total_samples = waveform.size(0)
 
@@ -213,14 +214,33 @@ def split_audio(waveform, sample_rate):
         if end <= total_samples:
             segment = waveform[start:end]
             segments.append(segment)
-    
-    # In case audio length is shorter than segment length.
-    if len(segments) == 0: 
-        segment = waveform
-        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)
+
+#     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)
+    
+#     # In case audio length is shorter than segment length.
+#     if len(segments) == 0: 
+#         segment = waveform
+#         segments.append(segment)
+
+#     return segments
+
 
 def safe_remove_dir(directory):
     """