Spaces:
Running
on
Zero
Running
on
Zero
tori29umai
commited on
Commit
•
c1c88a4
1
Parent(s):
b36fcf5
Update
Browse files
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(
|
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)}
|