wendys-llc commited on
Commit
18418b8
1 Parent(s): 7fdad91

Update app.py

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