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()