j commited on
Commit
caf6171
1 Parent(s): cc634a8

change indexing into sig_seg according to AudioSignal

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -58,7 +58,7 @@ def process_fn(input_audio_path, seed, guidance_scale, num_inference_steps):
58
  end = start + 5.12
59
 
60
  # get segment of audio from original file
61
- sig_seg = sig[int(start*sig.sample_rate):int(end*sig.sample_rate)] # int accounts for float end time on last seg
62
  print(f"Segment length: {sig_seg.duration}")
63
  print(f"Segment start: {start}")
64
  print(f"Segment end: {end}")
 
58
  end = start + 5.12
59
 
60
  # get segment of audio from original file
61
+ sig_seg = sig[..., int(start*sig.sample_rate):int(end*sig.sample_rate)] # int accounts for float end time on last seg
62
  print(f"Segment length: {sig_seg.duration}")
63
  print(f"Segment start: {start}")
64
  print(f"Segment end: {end}")