impresso-project
/

impresso-langident

Simon Clematide commited on 13 days ago

Commit

996ea64

1 Parent(s): 3b1ef7b

Fix confidence score rounding in model output for JSON compatibility

Files changed (1) hide show

langident.py CHANGED Viewed

@@ -25,5 +25,5 @@ class LangDetectionPipeline(Pipeline):
         )  # Convert to float for JSON serialization
         # Format as JSON-compatible dictionary
-        model_output = {"language": label, "score": round(confidence * 100, 2)}
         return model_output

         )  # Convert to float for JSON serialization
         # Format as JSON-compatible dictionary
+        model_output = {"language": label, "score": round(confidence, 2)}
         return model_output