up

Refound-445 · Oct 31, 2024 · bce2e7a · bce2e7a
1 parent 7999c41
commit bce2e7a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/nonebot_plugin_nailongremove/__init__.py b/nonebot_plugin_nailongremove/__init__.py
@@ -9,7 +9,7 @@
 from . import handler as handler
 from .config import Config
 
-__version__ = "2.2.0"
+__version__ = "2.2.0.post1"
 __plugin_meta__ = PluginMetadata(
     name="自动撤回奶龙",
     description="一个基于图像分类模型的简单插件~",

diff --git a/nonebot_plugin_nailongremove/model/target_detection.py b/nonebot_plugin_nailongremove/model/target_detection.py
@@ -80,7 +80,7 @@ def check_image(image: np.ndarray) -> "CheckResult":
     has = any(
         True
         for c, s in zip(final_cls_inds, final_scores)
-        if labels[c] == "nailong" and s >= config.nailong_model1_score
+        if labels[int(c)] == "nailong" and s >= config.nailong_model1_score
     )
     if has:
         image = vis(