import spaces 
import gradio as gr
from inference import predict

@spaces.GPU
def predict_from_audio(audio):
    return predict(audio)["language"]

demo = gr.Interface(
    fn=predict_from_audio,
    inputs=gr.Audio(type="filepath"),
    outputs="label",
    title="Language Identification Model",
    description="A Malaysian Language Identification Model for Detecting English, Mandarin, Tamil, Malay, Iban, and Bukar Sadong."
)

if __name__ == "__main__":
    demo.launch()