import torch
from torch.nn.utils.rnn import pad_sequence

def collate_fn(batch):
    inputs, labels = zip(*batch)  # Séparer les features et les labels
    inputs = pad_sequence(inputs, batch_first=True, padding_value=0)  # Padding des audios
    labels = torch.tensor(labels, dtype=torch.long)  # Conversion en tensor
    return inputs, labels