File size: 546 Bytes
1e3fde7 f7c8dec 1e3fde7 f499d66 1e3fde7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
import argparse
from api import classify_audio_clip
from tortoise.utils.audio import load_audio
if __name__ == '__main__':
parser = argparse.ArgumentParser()
parser.add_argument('--clip', type=str, help='Path to an audio clip to classify.', default="../examples/favorite_riding_hood.mp3")
args = parser.parse_args()
clip = load_audio(args.clip, 24000)
clip = clip[:, :220000]
prob = classify_audio_clip(clip)
print(f"This classifier thinks there is a {prob*100}% chance that this clip was generated from Tortoise.") |