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/tfjs")
|
||||||
gen.require("https://cdn.jsdelivr.net/npm/@tensorflow-models/toxicity")
|
gen.require("https://cdn.jsdelivr.net/npm/@tensorflow-models/toxicity")
|
||||||
let threshold = 0.9;
|
let threshold = 0.9;
|
||||||
|
let toxics = []
|
||||||
toxicity.load(threshold).then(model => {
|
toxicity.load(threshold).then(model => {
|
||||||
model.classify(sentences).then(predictions => {
|
model.classify(sentences).then(predictions => {
|
||||||
predictions.forEach(classified => {
|
predictions.forEach(classified => {
|
||||||
if (classified.label == "toxicity") {
|
toxics = classified.results.match
|
||||||
var toxic = classified.results.match
|
|
||||||
}
|
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
});
|
});
|
||||||
return toxic
|
return toxics;
|
||||||
}
|
}
|
||||||
|
|
||||||
async asyncSleep(ms) {
|
async asyncSleep(ms) {
|
||||||
|
|
Loading…
Reference in a new issue