Spaces:
Running
Running
File size: 2,691 Bytes
daf4e8e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
import os
import re
from io import BytesIO
from urllib.parse import urlparse
import requests
import torchaudio
def load_audio(source):
def is_url(path):
try:
result = urlparse(path)
return all([result.scheme, result.netloc])
except Exception:
return False
if is_url(source):
# 从 URL 加载音频
response = requests.get(source)
response.raise_for_status() # 检查请求状态
audio_data = BytesIO(response.content) # 转为类文件对象
else:
# 从本地文件加载音频
if not os.path.exists(source):
raise FileNotFoundError(f"File not found: {source}")
audio_data = source # 本地路径可以直接传递给 torchaudio.load
# 使用 torchaudio 加载音频
waveform, sample_rate = torchaudio.load(audio_data)
return waveform, sample_rate
pattern = re.compile(r"<\|speech-(\d+)\|>")
def combine_sequences(first_elements, second_elements, third_elements):
group_size = 7
sequence = []
second_index = 0
third_index = 0
for first in first_elements:
group = [None] * group_size
# Assign the first element
group[0] = first
# Assign the second and fifth elements if they exist
if second_index < len(second_elements):
group[1] = second_elements[second_index]
second_index += 1
if second_index < len(second_elements):
group[4] = second_elements[second_index]
second_index += 1
# Assign the remaining elements from third_elements if they exist
for j in [2, 3, 5, 6]:
if third_index < len(third_elements):
group[j] = third_elements[third_index]
third_index += 1
# Remove None values at the end of the group if the group is incomplete
sequence.extend([x for x in group if x is not None])
return sequence
def split_sequence(sequence):
group_size = 7
first_elements = []
second_elements = []
third_elements = []
# Iterate over the sequence in chunks of 7
for i in range(0, len(sequence), group_size):
group = sequence[i:i + group_size]
# Add elements to the respective lists based on their position in the group
if len(group) >= 1:
first_elements.append(group[0])
if len(group) >= 5:
second_elements.extend([group[1], group[4]])
if len(group) >= 7:
third_elements.extend([group[2], group[3], group[5], group[6]])
else:
third_elements.extend(group[2:])
return first_elements, second_elements, third_elements
|