Update jspp.js
This commit is contained in:
parent
24acc782d3
commit
37b034f067
1 changed files with 3 additions and 5 deletions
8
jspp.js
8
jspp.js
|
@ -18,17 +18,15 @@ class JSPlusPlus {
|
|||
gen.require("https://cdn.jsdelivr.net/npm/@tensorflow/tfjs")
|
||||
gen.require("https://cdn.jsdelivr.net/npm/@tensorflow-models/toxicity")
|
||||
let threshold = 0.9;
|
||||
|
||||
let toxics = []
|
||||
toxicity.load(threshold).then(model => {
|
||||
model.classify(sentences).then(predictions => {
|
||||
predictions.forEach(classified => {
|
||||
if (classified.label == "toxicity") {
|
||||
var toxic = classified.results.match
|
||||
}
|
||||
toxics = classified.results.match
|
||||
});
|
||||
});
|
||||
});
|
||||
return toxic
|
||||
return toxics;
|
||||
}
|
||||
|
||||
async asyncSleep(ms) {
|
||||
|
|
Loading…
Reference in a new issue