Spaces:
Sleeping
Sleeping
File size: 371 Bytes
c2864d3 fcf5834 c2864d3 fcf5834 c2864d3 |
1 2 3 4 5 6 7 8 9 10 11 12 |
from flask import Flask, request, jsonify
from transformers import pipeline
app = Flask(__name__)
classifier = pipeline("text-classification", model="j-hartmann/emotion-english-distilroberta-base", return_all_scores=True)
@app.route('/classify', methods=['POST'])
def classify():
text = request.json['text']
result = classifier(text)
return jsonify(result)
|