File size: 1,041 Bytes
4d03d1c c4668be 4d03d1c c4668be 4d03d1c fd718f0 4d03d1c | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 | import pickle # Pour sauvegarder et charger le modèle
# Gradio est une bibliothèque pour créer des interfaces utilisateur interactives
import gradio as gr
# Charger le modèle sauvegardé
with open('sms_spam_detector_model_fr.pkl','rb') as file:
model=pickle.load(file)
# Charger le vectorizer
with open('sms_vectorizer_fr.pkl','rb') as file:
vectorizer=pickle.load(file)
# Fonction pour prédire si un message est du spam ou non
def predict_spam(message):
# Transformer le message en utilisant le vectorizer
message_transformed = vectorizer.transform([message])
prediction = model.predict(message_transformed)
return "SPAM ❌" if prediction[0] == 1 else "HAM ✅"
# Créer une interface Gradio pour l'application
interface_gradio = gr.Interface(fn=predict_spam, inputs="text", outputs="text",
title="Sms Spam Detector",
description="print message to detect spam or ham."
)
interface_gradio.launch() |