Myspace2 / app.py
Dmitry72's picture
Create app.py
04dc129
raw
history blame
327 Bytes
import json
import gradio as gr
def image_classifier(inp):
pipe = pipeline(task='text-classification', model='s-nlp/russian_toxicity_classifier')
out = pipe(inp)
a = json.dumps(out[0])
return a
demo = gr.Interface(fn=image_classifier, inputs="text", outputs="label")
demo.launch(server_port=11459, debug=True)