Lord-Raven commited on
Commit
2a07486
·
1 Parent(s): 249b712

Trying zero-shot classification.

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ def zero_shot_classification(data_string):
8
  data = json.loads(data_string)
9
  print(data)
10
  classifier = pipeline(task='zero-shot-classification', model='typeform/distilbert-base-uncased-mnli')
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
 
 
8
  data = json.loads(data_string)
9
  print(data)
10
  classifier = pipeline(task='zero-shot-classification', model='typeform/distilbert-base-uncased-mnli')
11
+ print('classifier setup')
12
  results = classifier(data['sequence'], candidate_labels=data['candidate_labels'], hypothesis_template=data['hypothesis_template'], multi_label=data['multi_label'])
13
  return results
14