spam-detector / app.py
sdafd's picture
Update app.py
e6281e2 verified
raw
history blame
470 Bytes
import gradio as gr
from transformers import pipeline
def analyze_output(input_text):
pipe = pipeline("text-classification", model="ZachBeesley/Spam-Detector")
result = pipe(input_text)[0]
return str(result)
iface = gr.Interface(
fn=analyze_output,
inputs=gr.Textbox(),
outputs="text",
live=False,
title="Spam Classification App",
description="Enter a text to classify it as Spam or Not Spam.",
theme="compact"
)
iface.launch()