tori29umai commited on
Commit
c1c88a4
1 Parent(s): b36fcf5
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -59,7 +59,7 @@ def main(image_path, model_id):
59
  NSFW_flag, IP_flag, tag_text = evaluate_tags(prob, rating_tags, character_tags, general_tags)
60
  return NSFW_flag, IP_flag, tag_text
61
 
62
- def evaluate_tags(sprob, rating_tags, character_tags, general_tags):
63
  thresh = 0.35
64
  # NSFW/SFW判定
65
  tag_confidences = {tag: prob[i] for i, tag in enumerate(rating_tags)}
 
59
  NSFW_flag, IP_flag, tag_text = evaluate_tags(prob, rating_tags, character_tags, general_tags)
60
  return NSFW_flag, IP_flag, tag_text
61
 
62
+ def evaluate_tags(prob, rating_tags, character_tags, general_tags):
63
  thresh = 0.35
64
  # NSFW/SFW判定
65
  tag_confidences = {tag: prob[i] for i, tag in enumerate(rating_tags)}