Spaces:
Runtime error
Runtime error
File size: 702 Bytes
7785262 39795a3 7785262 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
from transformers import TextClassificationPipeline, BertForSequenceClassification, AutoTokenizer
model_name = 'smilegate-ai/kor_unsmile'
model = BertForSequenceClassification.from_pretrained(model_name)
tokenizer = AutoTokenizer.from_pretrained(model_name)
pipe = TextClassificationPipeline(
model = model,
tokenizer = tokenizer,
device = -1, # cpu: -1, gpu: gpu number
return_all_scores = True,
function_to_apply = 'sigmoid'
)
from flask import Flask, jsonify, request
app = Flask(__name__)
@app.route('/detect', methods=["POST"])
def detect():
text = request.json["text"]
return jsonify(pipe(text))
if __name__ == "__main__":
app.run('0.0.0.0', 7860, False)
|