document.getElementById("generateBtn").onclick = async function(){

    const prompt = document.getElementById("prompt").value;

    const task = detectTask(prompt);

    let modelName = MODELS[task];

    if(!modelName){
        modelName = MODELS.text;
    }

    document.getElementById("model").innerHTML = modelName;

    document.getElementById("output").innerHTML = "Generating...";

    const result = await queryHF(modelName,prompt);

    if(Array.isArray(result)){

        if(result[0].generated_text){

            document.getElementById("output").innerHTML =
                result[0].generated_text;

        }else{

            document.getElementById("output").innerHTML =
                JSON.stringify(result);

        }

    }else{

        document.getElementById("output").innerHTML =
            JSON.stringify(result,null,2);

    }

}