from flask import Flask, request, jsonify | |
from transformers import pipeline | |
app = Flask(__name__) | |
model = pipeline("sentiment-analysis") | |
def predict_sentiment(): | |
text = request.json.get('text') | |
if not text: | |
return jsonify({'error': 'Text field is missing'}), 400 | |
result = model(text) | |
return jsonify({'sentiment': result[0]['label'], 'confidence': result[0]['score']}) | |
if __name__ == '__main__': | |
app.run(debug=True, host='0.0.0.0') |