Lord-Raven commited on
Commit
abbb119
·
1 Parent(s): 2d54881

Trying zero-shot classification.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 {results}
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,