| 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') |