import gradio as gr
from transformers import pipeline

pipeline = pipeline("text-classification", model="nickprock/distilbert-base-uncased-banking77-classification")

def predict(text):
  predictions = pipeline(text)
  return {p["label"]: p["score"] for p in predictions}
  
  gr.Interface(
    predict,
    inputs="textbox",
    outputs="label",
    theme="huggingface", 
    title="Banking Intent Classifier",
    description="Try to classify customer queries",
    examples=[["I can't pay by my credit card"],["The amount on the transfer is wrong can I cancel it?"]],
    share=True
).launch()