jorgefio commited on
Commit
2d139eb
β€’
1 Parent(s): 21dd98d

Rename zero-shot-classification.py to app.py

Browse files
zero-shot-classification.py β†’ app.py RENAMED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
 
4
  classifier = pipeline("zero-shot-classification")
5
- labels = ["Connectivity", "Billing", "Setup"]
6
 
7
  def predict(text):
8
  preds = classifier(text, candidate_labels=labels)
 
2
  from transformers import pipeline
3
 
4
  classifier = pipeline("zero-shot-classification")
5
+ labels = ["No Connectivity", "Billing", "Setup", "Slow Connection"]
6
 
7
  def predict(text):
8
  preds = classifier(text, candidate_labels=labels)