Change model output name from logits to audio

[will-rice]

This change makes sense because the audio, not the logits, is returned in the model output classes.