Increase model threshold and conservativeness

functions/model_infer.py

@@ -37,5 +37,6 @@ def preprocess(sentences):
 def predict_from_document(sentences):
     preprop = preprocess(sentences)
     prediction = model.predict(preprop)
-    output = (prediction.flatten()[: len(sentences)] >= 0.5).astype(int)
+    # Set the prediction threshold to 0.8
+    output = (prediction.flatten()[: len(sentences)] >= 0.8).astype(int)
     return output, prediction.flatten()[: len(sentences)]