mirror of
https://github.com/storytold/storyteller-ml.git
synced 2026-10-23 08:26:06 +00:00
fix bugs introduced during slicing
This commit is contained in:
@@ -79,13 +79,14 @@ def split_silence(
|
|||||||
yield Chunk(
|
yield Chunk(
|
||||||
is_speech=False, audio=audio[last_end:start], start=last_end, end=start
|
is_speech=False, audio=audio[last_end:start], start=last_end, end=start
|
||||||
)
|
)
|
||||||
if end-start > chunk_size_max:
|
if end-start > chunk_size_max:
|
||||||
while last_end < end:
|
chunk_count = (end-start) // chunk_size_max
|
||||||
yield Chunk(is_speech=True, audio=audio[last_end:last_end+chunk_size_max], start=last_end, end=last_end+chunk_size_max)
|
for i in range(0, chunk_count):
|
||||||
last_end = last_end+chunk_size_max
|
yield Chunk(is_speech=True, audio=audio[start+(i*chunk_size_max):start+((i+1)*chunk_size_max)],
|
||||||
else:
|
start=start+(i*chunk_size_max), end=start+((i+1)*chunk_size_max))
|
||||||
yield Chunk(is_speech=True, audio=audio[start:end], start=start, end=end)
|
yield Chunk(is_speech=True, audio=audio[start+chunk_count*chunk_size_max:end],
|
||||||
last_end = end
|
start=start+chunk_count*chunk_size_max, end=end)
|
||||||
|
last_end=end
|
||||||
else:
|
else:
|
||||||
yield Chunk(is_speech=True, audio=audio[start:end], start=start, end=end)
|
yield Chunk(is_speech=True, audio=audio[start:end], start=start, end=end)
|
||||||
last_end = end
|
last_end = end
|
||||||
|
|||||||
Reference in New Issue
Block a user