Spaces:
Runtime error
Runtime error
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() |