Spaces:
Sleeping
Sleeping
File size: 614 Bytes
61d03da 92f34f9 61d03da 34aed82 1458121 61d03da |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
import gradio as gr
from transformers import pipeline
def analyze_output(input_text):
pipe = pipeline("text-classification", model="Titeiiko/OTIS-Official-Spam-Model")
x = pipe(input)[0]
if x["label"] == "LABEL_0":
return {"type":"Not Spam", "probability":x["score"]}
else:
return {"type":"Spam", "probability":x["score"]}
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()
|