nickprock's picture
Update app.py
629a40f
raw
history blame
586 Bytes
import gradio as gr
from transformers import pipeline
pipe = pipeline("text-classification", model="nickprock/distilbert-base-uncased-banking77-classification")
def greet(text):
response = pipe(text)
return response[0]['label']
demo = gr.Interface(fn=greet,
inputs="text",
outputs="text",
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?"], "My card still hasn't been delivered "],
share=True)
demo.launch()