tortoise/is_this_from_tortoise.py

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."
    )