Update

app.py

@@ -56,10 +56,10 @@ def main(image_path, model_id):
     general_tags = [row[1] for row in rows if row[2] == "0"]
 
     # タグと評価
-    NSFW_flag, IP_flag, tag_text = self.evaluate_tags(prob, rating_tags, character_tags, general_tags)
+    NSFW_flag, IP_flag, tag_text = evaluate_tags(prob, rating_tags, character_tags, general_tags)
     return NSFW_flag, IP_flag, tag_text
 
-def evaluate_tags(self, prob, rating_tags, character_tags, general_tags):
+def evaluate_tags(sprob, rating_tags, character_tags, general_tags):
     thresh = 0.35
     # NSFW/SFW判定
     tag_confidences = {tag: prob[i] for i, tag in enumerate(rating_tags)}