Update fn.py

fn.py

@@ -53,7 +53,7 @@ def check(image):
     prob = ort_sess.run(None, {input_name: img})[0][0]  # ONNXモデルからの出力
 
     # NSFW/SFW判定
-    tag_confidences = {tag: prob[i] for i, tag in enumerate(rating_tags)}
+    tag_confidences = {tag: float(prob[i]) for i, tag in enumerate(rating_tags)}
     max_nsfw_score = max(tag_confidences.get("questionable", 0), tag_confidences.get("explicit", 0))
     max_sfw_score = tag_confidences.get("general", 0)
 
@@ -73,7 +73,7 @@ def check(image):
             tag_index = i - 4
             if tag_index < len(general_tags):
                 tag_name = general_tags[tag_index]
-                general_tags_with_probs[tag_name] = p
+                general_tags_with_probs[tag_name] = float(p)
     tags = sorted(general_tags_with_probs.items(), key = lambda x : x[1], reverse=True)
 
     return {