make result more helpful

newhinton · Apr 28, 2024 · f1dfca6 · f1dfca6
1 parent 004c327
commit f1dfca6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/checkProfanity.py b/scripts/checkProfanity.py
@@ -11,7 +11,7 @@
     print("Checking: "+str(line.rstrip()))
     prediction = predict_prob([line.rstrip()])
     if prediction[0] > 0.5:
-      file.write(str(line.rstrip())+"\n")
+      file.write(str(prediction[0]) + " - " + str(line.rstrip())+"\n")
       print("Offending line: "+str(line.rstrip()))
 
 file.close()