Spaces:
Sleeping
Sleeping
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)
|