Northern_Lights/ZeroBot-Plugin
1
0
Fork 0
mirror of https://github.com/FloatTech/ZeroBot-Plugin.git synced 2026-02-12 10:10:25 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

🚧 move tag & nsfw to azfun

Browse Source
This commit is contained in:
源文雨
2022-04-05 15:52:06 +08:00
parent 56e53b75a5
commit d8a429d3b3
4 changed files with 10 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
plugin/nsfw/main.go
View File

@@ -23,12 +23,12 @@ func init() {
url := ctx.State["image_url"].([]string)
if len(url) > 0 {
ctx.SendChain(message.Text("少女祈祷中..."))
p, err := nsfw.Classify(url...)
p, err := nsfw.Classify(url[0])
if err != nil {
ctx.SendChain(message.Text("ERROR:", err))
return
}
ctx.Send(message.ReplyWithMessage(ctx.Event.MessageID, message.Text(judge(p[0]))))
ctx.Send(message.ReplyWithMessage(ctx.Event.MessageID, message.Text(judge(p))))
}
})
control.Register("nsfwauto", &control.Options{
@@ -39,17 +39,17 @@ func init() {
url := ctx.State["image_url"].([]string)
if len(url) > 0 {
process.SleepAbout1sTo2s()
p, err := nsfw.Classify(url...)
p, err := nsfw.Classify(url[0])
if err != nil {
return
}
process.SleepAbout1sTo2s()
autojudge(ctx, p[0])
autojudge(ctx, p)
}
})
}
func judge(p nsfw.Picture) string {
func judge(p *nsfw.Picture) string {
if p.Neutral > 0.3 {
return "普通哦"
}
@@ -71,7 +71,7 @@ func judge(p nsfw.Picture) string {
return c
}
func autojudge(ctx *zero.Ctx, p nsfw.Picture) {
func autojudge(ctx *zero.Ctx, p *nsfw.Picture) {
if p.Neutral > 0.3 {
return
}

2
plugin/scale/main.go
View File

@@ -56,7 +56,7 @@ func init() {
ctx.SendChain(message.Text("ERROR:", err))
return
}
if p[0].Drawings < 0.1 || p[0].Neutral > 0.8 {
if p.Drawings < 0.1 || p.Neutral > 0.8 {
ctx.SendChain(message.Text("请发送二次元图片!"))
return
}