from transformers import pipeline as pl
import gradio as gr

oracle = pl("text-classification")

def pipe(text: str):
    return oracle(text)


interface = gr.Interface(pipe, gr.Textbox(label="Prompt"), gr.Textbox(label="Response"), title="Text Classifier")

if __name__ == "__main__":
    interface.launch()