Spaces:
Sleeping
Sleeping
File size: 511 Bytes
f085882 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 | import gradio as gr
import joblib
model = joblib.load("model.pkl")
vectorizer = joblib.load("vectorizer.pkl")
def predict(text):
X = vectorizer.transform([text])
pred = model.predict(X)[0]
return "🚨 SPAM" if pred == 1 else "✅ NOT SPAM"
iface = gr.Interface(
fn=predict,
inputs=gr.Textbox(label="Enter a message"),
outputs=gr.Label(label="Prediction"),
title="Spam Detector",
description="A simple ML model that classifies messages as spam or not spam."
)
iface.launch()
|