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,13 +18,12 @@ 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 toxic = false
toxicity.load(threshold).then(model => {
model.classify(sentences).then(predictions => {
predictions.forEach(classified => {
if (classified.label == "toxicity") {
toxic = classified.results.match
var toxic = classified.results.match
}
});
});