import gradio as gr
from transformers import pipeline

pipe = pipeline("text-classification", model="marvinisjarvis/model")
def predict(text):
    result = pipe(text)[0]  # take first element of list
    return {result['label']: float(result['score'])}

demo = gr.Interface(fn=predict, inputs="text", outputs="label")
if __name__ == "__main__":
    demo.launch()