nickprock's picture
create app.py
d827478
raw
history blame
580 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?"]]
).launch()