Fix split long audio bug
This commit is contained in:
@@ -101,9 +101,12 @@ class WhisperComputeNode(ComputeNode):
|
|||||||
text = ""
|
text = ""
|
||||||
results = []
|
results = []
|
||||||
latest_resp = None
|
latest_resp = None
|
||||||
step = 30 * 1000
|
step = 10 * 60 * 1000
|
||||||
for i in range(0, 60 * 1000, step):
|
for i in range(0, len(audio), step):
|
||||||
chunk = audio[i:i + step]
|
if i + step < len(audio):
|
||||||
|
chunk = audio[i:i + step]
|
||||||
|
else:
|
||||||
|
chunk = audio[i:]
|
||||||
seg_file = io.BytesIO()
|
seg_file = io.BytesIO()
|
||||||
chunk.export(seg_file, format="mp3")
|
chunk.export(seg_file, format="mp3")
|
||||||
seg_file.seek(0)
|
seg_file.seek(0)
|
||||||
|
|||||||
Reference in New Issue
Block a user