Update app.py

app.py

@@ -157,30 +157,30 @@ def resample_waveform(waveform, original_sample_rate, target_sample_rate):
 
 
 
-def split_audio(waveform, sample_rate):
-    segment_samples = segment_duration * sample_rate
-    total_samples = waveform.size(0)
+# def split_audio(waveform, sample_rate):
+#     segment_samples = segment_duration * sample_rate
+#     total_samples = waveform.size(0)
     
-    segments = []
-    # If the audio is shorter than the segment duration, just use the entire audio
-    if total_samples <= segment_samples:
-        segments.append(waveform)
-    else:
-        # Split the audio into segments of the specified duration
-        for start in range(0, total_samples, segment_samples):
-            end = min(start + segment_samples, total_samples)
-            segment = waveform[start:end]
-            segments.append(segment)
+#     segments = []
+#     # If the audio is shorter than the segment duration, just use the entire audio
+#     if total_samples <= segment_samples:
+#         segments.append(waveform)
+#     else:
+#         # Split the audio into segments of the specified duration
+#         for start in range(0, total_samples, segment_samples):
+#             end = min(start + segment_samples, total_samples)
+#             segment = waveform[start:end]
+#             segments.append(segment)
     
-    # Ensure we have at least one segment with a minimum length
-    if len(segments) == 0 or all(len(segment) < 100 for segment in segments):
-        # Create a padded segment if audio is too short
-        padded_segment = torch.zeros(segment_samples)
-        if total_samples > 0:
-            padded_segment[:total_samples] = waveform
-        segments = [padded_segment]
+#     # Ensure we have at least one segment with a minimum length
+#     if len(segments) == 0 or all(len(segment) < 100 for segment in segments):
+#         # Create a padded segment if audio is too short
+#         padded_segment = torch.zeros(segment_samples)
+#         if total_samples > 0:
+#             padded_segment[:total_samples] = waveform
+#         segments = [padded_segment]
         
-    return segments
+#     return segments
 
 # def split_audio(waveform, sample_rate):
 #     segment_samples = segment_duration * sample_rate
@@ -203,23 +203,23 @@ def split_audio(waveform, sample_rate):
 
 #     return segments
 
-# def split_audio(waveform, sample_rate):
-#     segment_samples = segment_duration * sample_rate
-#     total_samples = waveform.size(0)
+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)
+    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)
+    # In case audio length is shorter than segment length.
+    if len(segments) == 0: 
+        segment = waveform
+        segments.append(segment)
 
-#     return segments
+    return segments
 
 
 def safe_remove_dir(directory):