Update jspp.js

This commit is contained in:
Sam Sneed 2024-05-31 22:43:43 +00:00 committed by GitHub
parent 3dffa7e412
commit 24acc782d3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -18,18 +18,17 @@ 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 toxic = false
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") { if (classified.label == "toxicity") {
toxic = classified.results.match var toxic = classified.results.match
} }
}); });
}); });
}); });
return toxic return toxic
} }
async asyncSleep(ms) { async asyncSleep(ms) {