-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathapp.py
More file actions
22 lines (18 loc) · 680 Bytes
/
app.py
File metadata and controls
22 lines (18 loc) · 680 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
from flask import Flask, render_template, request
import joblib
# Load model and vectorizer
model = joblib.load("svm_spam_classifier.pkl")
vectorizer = joblib.load("tfidf_vectorizer.pkl")
app = Flask(__name__)
@app.route("/", methods=["GET", "POST"])
def index():
prediction = None
message = ""
if request.method == "POST":
message = request.form["email"]
vector = vectorizer.transform([message])
pred = model.predict(vector)[0]
prediction = "Spam" if pred == 1 else "Ham"
return render_template("index.html", prediction=prediction, message=message)
if __name__ == "__main__":
app.run(host='0.0.0.0', port=5000, debug=True)