nickprock's picture
Update app.py
249ccc6
raw
history blame
596 Bytes
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()