Update app.py

app.py

@@ -1,19 +1,19 @@
 import gradio as gr
 from transformers import pipeline
 
-pipeline = pipeline("text-classification", model="nickprock/distilbert-base-uncased-banking77-classification")
+pipe = 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=gr.outputs.Label(num_top_classes=1),
-    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?"]],
-    share=True
-).launch()
+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()