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