File size: 327 Bytes
04dc129
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
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)