Spaces:
Running
on
Zero
Running
on
Zero
Lord-Raven
commited on
Commit
·
6deae7c
1
Parent(s):
d04fba3
Trying zero-shot classification.
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio
|
|
2 |
from transformers import pipeline
|
3 |
|
4 |
def zero_shot_classification(data):
|
5 |
-
classifier = pipeline('zero-shot-classification, model='Xenova/mobilebert-uncased-mnli')
|
6 |
|
7 |
results = classifier(data.sequence, candidate_labels=data.candidate_labels, hypothesis_template=data.hypothesis_template, multi_label=data.multi_label)
|
8 |
return {results}
|
|
|
2 |
from transformers import pipeline
|
3 |
|
4 |
def zero_shot_classification(data):
|
5 |
+
classifier = pipeline('zero-shot-classification', model='Xenova/mobilebert-uncased-mnli')
|
6 |
|
7 |
results = classifier(data.sequence, candidate_labels=data.candidate_labels, hypothesis_template=data.hypothesis_template, multi_label=data.multi_label)
|
8 |
return {results}
|