Spaces:
Runtime error
Runtime error
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__) | |
def detect(): | |
text = request.json["text"] | |
return jsonify(pipe(text)) | |
if __name__ == "__main__": | |
app.run('0.0.0.0', 7860, False) | |