Spaces:

e-eeeema
/

intrusion-detection

Sleeping

e-eeeema commited on Mar 21

Commit

05644bb

verified ·

1 Parent(s): 88c3ea9

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -18,7 +18,9 @@ def predict():
   # Apply threshold 0.3
   prediction = int(probs[0] > 0.3)
-  return jsonify({'attack_detected': prediction})
 if __name__ == '__main__':
   app.run(host='0.0.0.0', port=7860)

   # Apply threshold 0.3
   prediction = int(probs[0] > 0.3)
+  return jsonify({'attack_detected': prediction,
+                 'probability': float(probs[0])
+                 })
 if __name__ == '__main__':
   app.run(host='0.0.0.0', port=7860)