File size: 860 Bytes
47d13ac
516742a
47d13ac
516742a
47d13ac
516742a
47d13ac
516742a
47d13ac
 
516742a
 
47d13ac
516742a
47d13ac
516742a
47d13ac
516742a
47d13ac
516742a
47d13ac
516742a
47d13ac
 
516742a
47d13ac
516742a
47d13ac
 
516742a
47d13ac
516742a
47d13ac
516742a
47d13ac
 
516742a
 
3af0a5b
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
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);

    }

}