| from flask import Flask, request, render_template |
| import joblib |
|
|
| app = Flask(__name__) |
| model = joblib.load(open("model/sentiment_model.pkl", "rb")) |
| vectorizer = joblib.load(open("model/vectorizer.pkl", "rb")) |
|
|
| @app.route('/') |
| def home(): |
| return render_template('index.html') |
|
|
| @app.route('/predict', methods=['POST']) |
| def predict(): |
| text = request.form['user_text'] |
| text_vector = vectorizer.transform([text]) |
| prediction = model.predict(text_vector)[0] |
| proba = model.predict_proba(text_vector) |
| sentiment = prediction |
| return render_template('index.html', sentiment=sentiment, neg_proba=proba[0][0], posi_proba=proba[0][1], |
| user_input=text) |
|
|
| if __name__ == '__main__': |
| app.run(host="0.0.0.0", port=8080, debug=True) |
|
|