from transformers import pipeline
import gradio as gr

model = pipeline(model="seara/rubert-tiny2-ru-go-emotions")

def text_classification(data):
  result = model(data)[0]['label']
  return result

demo = gr.Interface(fn=text_classification, inputs="textbox", outputs="textbox")
    
demo.launch(share=True)