import gradio as gr
from predict import prediction

title = "Audio deepfake Classification using 2s segment mel-spectrograms with Examples Only"
description = """This space uses fine-tuned kubinooo/convnext-tiny-224-audio-deepfake-classification model to classify audio files.
"""

demo = gr.Interface(
    title=title,
    inputs=gr.Audio( type="filepath",
        interactive=True,  # This prevents users from uploading their own images
        show_label=True,
        label="Select from examples below or upload/record your own audio"),
    fn=prediction,
    outputs=gr.Label(
        num_top_classes=2,
    ),
    examples=[
        "src/31-MALE-VC-FAKE.wav",
        "src/2152-REAL.wav"
    ],
    description=description
)

demo.launch(share=True)