| from transformers import pipeline | |
| 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(share=True) |