+3
-3
automod/visual/hiveai_client.go
+3
-3
automod/visual/hiveai_client.go
···
60
60
61
61
for _, cls := range cl {
62
62
if cls.Class == "very_bloody" && cls.Score >= 0.90 {
63
-
labels = append(labels, "gore")
63
+
labels = append(labels, "graphic-media")
64
64
}
65
65
if cls.Class == "human_corpse" && cls.Score >= 0.90 {
66
-
labels = append(labels, "corpse")
66
+
labels = append(labels, "graphic-media")
67
67
}
68
68
if cls.Class == "hanging" && cls.Score >= 0.90 {
69
-
labels = append(labels, "corpse")
69
+
labels = append(labels, "graphic-media")
70
70
}
71
71
if cls.Class == "yes_self_harm" && cls.Score >= 0.96 {
72
72
labels = append(labels, "self-harm")