Update app.py
Browse files
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 |
-
|
|
|
|
|
|
|
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)
|