spam-detector / app.py
sdafd's picture
Update app.py
005f72b verified
import gradio as gr
from transformers import pipeline
def analyze_output(input_text):
pipe = pipeline("text-classification", model="Titeiiko/OTIS-Official-Spam-Model")
result = pipe(input_text)[0]
if result["label"] == "LABEL_0":
return f"Not Spam: {result['score']:.4f}"
else:
return f"Spam: {result['score']:.4f}"
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()