zerospam / app.py
kollera's picture
Create app.py
5f7a34d verified
raw
history blame
595 Bytes
import gradio as gr
from transformers import pipeline
# Inizializza il modello DistilBERT per la classificazione di testo
classifier = pipeline("text-classification", model="distilbert-base-uncased")
def classify_email(text):
result = classifier(text)
return result
# Interfaccia con Gradio
iface = gr.Interface(fn=classify_email,
inputs="text",
outputs="label",
title="ZeroSpam Email Classifier",
description="Inserisci l'email da analizzare per determinare se è spam o phishing.")
iface.launch()