speech-to-text / app.py
Nechba's picture
Update app.py
c2864d3 verified
raw
history blame
371 Bytes
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)