rmdhirr commited on
Commit
531c628
1 Parent(s): 12cad08

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -79,10 +79,10 @@ def get_prediction(input_text, input_type):
79
 
80
  def phishing_detection(input_text, input_type):
81
  prediction = get_prediction(input_text, input_type)
82
- if prediction > 0.5:
83
- return f"Warning: This site is likely a phishing site! ({prediction:.2f})"
84
  else:
85
- return f"Safe: This site is not likely a phishing site. ({prediction:.2f})"
86
 
87
  iface = gr.Interface(
88
  fn=phishing_detection,
 
79
 
80
  def phishing_detection(input_text, input_type):
81
  prediction = get_prediction(input_text, input_type)
82
+ if prediction > 0.7:
83
+ return f"Warning: This site is likely a phishing site!"
84
  else:
85
+ return f"Safe: This site is not likely a phishing site."
86
 
87
  iface = gr.Interface(
88
  fn=phishing_detection,