File size: 580 Bytes
d827478
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
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()