lg3394 commited on
Commit
d4cf7bb
·
verified ·
1 Parent(s): 7b99567

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ classifier = pipeline("zero-shot-classification",
5
  model="facebook/bart-large-mnli")
6
 
7
  def do_action(text):
8
- candidate_labels = ['travel', 'cooking', 'dancing']
9
  result = classifier(text, candidate_labels)
10
 
11
  return result
 
5
  model="facebook/bart-large-mnli")
6
 
7
  def do_action(text):
8
+ candidate_labels = ['gun control', 'abortion', 'gender transition', 'freedom of speech', 'immigration', 'taxes']
9
  result = classifier(text, candidate_labels)
10
 
11
  return result