wendys-llc commited on
Commit
56db43e
·
1 Parent(s): 2124b87

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,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['categories'][0]
12
 
13
  iface = gr.Interface(fn=do_action, inputs="text", outputs="text")
14
  iface.launch()
 
8
  candidate_labels = ['gun control', 'abortion', 'gender transition', 'freedom of speech', 'immigration', 'taxes']
9
  result = classifier(text, candidate_labels)
10
 
11
+ return result['labels'][0]
12
 
13
  iface = gr.Interface(fn=do_action, inputs="text", outputs="text")
14
  iface.launch()