import gradio as gr
from inference_kathbadh import inference_kathbadh

audio_1 = gr.Audio(sources="upload", type="filepath", label="Audio 1")
audio_2 = gr.Audio(sources="upload", type="filepath", label="Audio 2")
text_output = gr.Textbox(label="Similarity Score")
gr.Interface(
    fn=inference_kathbadh,
    inputs=[audio_1, audio_2],
    outputs=text_output,
    title="Speaker Verification",
    description = "Speaker Verification using WavLM large model finetuned on Kathbadh dataset."
).launch()