kollera commited on
Commit
1a235fd
·
verified ·
1 Parent(s): b926b24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
app.py CHANGED
@@ -6,7 +6,10 @@ classifier = pipeline("text-classification", model="distilbert-base-uncased")
6
 
7
  def classify_email(text):
8
  result = classifier(text)
9
- return result
 
 
 
10
 
11
  # Interfaccia con Gradio
12
  iface = gr.Interface(fn=classify_email,
@@ -15,4 +18,4 @@ iface = gr.Interface(fn=classify_email,
15
  title="ZeroSpam Email Classifier",
16
  description="Inserisci l'email da analizzare per determinare se è spam o phishing.")
17
 
18
- iface.launch()
 
6
 
7
  def classify_email(text):
8
  result = classifier(text)
9
+ # `result` è una lista con un solo dizionario, prendiamo il primo elemento e restituiamo solo la label
10
+ label = result[0]['label']
11
+ score = result[0]['score']
12
+ return {label: score}
13
 
14
  # Interfaccia con Gradio
15
  iface = gr.Interface(fn=classify_email,
 
18
  title="ZeroSpam Email Classifier",
19
  description="Inserisci l'email da analizzare per determinare se è spam o phishing.")
20
 
21
+ iface.launch(share=True)