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