Upload 6 files

main.py

@@ -0,0 +1,43 @@
+import numpy as np
+import pandas as pd
+from sklearn.metrics import confusion_matrix
+from flask import Flask, request, render_template
+from tensorflow import keras
+from tensorflow.keras.preprocessing.text import Tokenizer
+from tensorflow.keras.preprocessing.sequence import pad_sequences
+from tensorflow.keras.models import Sequential
+from tensorflow.keras.layers import Embedding, Flatten, Dense
+import pickle
+
+app = Flask(__name__)
+
+# Load the pre-trained model and artifacts
+loaded_model = keras.models.load_model("spam_detection_model.h5")
+with open("tokenizer.pkl", "rb") as f:
+    loaded_tokenizer = pickle.load(f)
+with open("max_len.pkl", "rb") as f:
+    loaded_max_len = pickle.load(f)
+
+# Function to predict ham or spam using the pre-trained model
+def predict_message(message, model, tokenizer, max_len):
+    # Preprocess the message
+    sequence = tokenizer.texts_to_sequences([message])
+    padded_sequence = pad_sequences(sequence, maxlen=max_len, padding='post')
+    # Predict probabilities
+    probabilities = model.predict(padded_sequence)
+    # Convert probabilities to class labels
+    prediction = "Ham" if probabilities[0] < 0.5 else "Spam"
+    return prediction
+
+@app.route('/', methods=['GET', 'POST'])
+def home():
+    if request.method == 'POST':
+        # Get the message from the form input
+        message = request.form['message']
+        # Predict using the pre-trained model
+        prediction = predict_message(message, loaded_model, loaded_tokenizer, loaded_max_len)
+        return render_template('index.html', prediction=prediction)
+    return render_template('index.html')
+
+if __name__ == '__main__':
+    app.run(debug=True)

max_len.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:41da2fb4c152221cdaeec24a77d948d12cd6bd54cdd6433fc565bc5c2877ba77
+size 5

spam_alert.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c87ab31f58ef5b221d95c6f3d037857307fc7fb971d1d72a2876a140e6dae24f
+size 16163741

spam_detection_model.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d8364f593d378f8a4b90b703ce3f5318fc81310bdae1d1c120464dad12b95096
+size 16169896

tokenizer.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:bd6500f295b774cfb0e8848dc6a3b0fe968344237ae1f532b3ea6f9637cee658
+size 347210