Spaces:

ljh838
/

ace

Runtime error

ace / app.py

Update app.py

39795a3 over 1 year ago

No virus

702 Bytes

	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)