Spaces:
Running
on
Zero
Running
on
Zero
Lord-Raven
commited on
Commit
·
abbb119
1
Parent(s):
2d54881
Trying zero-shot classification.
Browse files
app.py
CHANGED
@@ -10,7 +10,7 @@ def zero_shot_classification(data_string):
|
|
10 |
classifier = pipeline(task='zero-shot-classification', tokenizer=AutoTokenizer.from_pretrained('sileod/deberta-v3-base-tasksource-nli'), model='sileod/deberta-v3-base-tasksource-nli')
|
11 |
|
12 |
results = classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])
|
13 |
-
return
|
14 |
|
15 |
gradio_interface = gradio.Interface(
|
16 |
fn = zero_shot_classification,
|
|
|
10 |
classifier = pipeline(task='zero-shot-classification', tokenizer=AutoTokenizer.from_pretrained('sileod/deberta-v3-base-tasksource-nli'), model='sileod/deberta-v3-base-tasksource-nli')
|
11 |
|
12 |
results = classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])
|
13 |
+
return results
|
14 |
|
15 |
gradio_interface = gradio.Interface(
|
16 |
fn = zero_shot_classification,
|