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)