rcraja03's picture
Upload 9 files
8a04130 verified
raw
history blame contribute delete
749 Bytes
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)