benjika commited on
Commit
8a8eac3
·
verified ·
1 Parent(s): 1298759

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -6,14 +6,14 @@ classifier = pipeline("text-classification", model="./model")
6
 
7
  def predict(text):
8
  result = classifier(text)
9
- return "Safe Email" if result[0]['label'] == "LABEL_0" else "Spam Email"
10
 
11
  iface = gr.Interface(
12
  fn=predict,
13
  inputs="text",
14
  outputs="label",
15
- title="Spam or not",
16
- description="Enter Email body to classify it as spam or not spam.",
17
  )
18
 
19
  iface.launch()
 
6
 
7
  def predict(text):
8
  result = classifier(text)
9
+ return "Safe Email" if result[0]['label'] == "LABEL_0" else "Phishing Email"
10
 
11
  iface = gr.Interface(
12
  fn=predict,
13
  inputs="text",
14
  outputs="label",
15
+ title="Spam or Phishing",
16
+ description="Enter Email body to classify it as spam or phishing.",
17
  )
18
 
19
  iface.launch()