kaushikbar commited on
Commit
d88be3e
1 Parent(s): 24bcb3c

added multi-label as a radio button user input

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -171,7 +171,8 @@ def sequence_to_classify(sequence, labels, multi_label):
171
  predicted_labels))
172
 
173
  if not multi_label:
174
- clean_output = {clean_output.keys()[0]: clean_output[clean_output.keys()[0]]}
 
175
  return clean_output
176
 
177
  iface = gr.Interface(
 
171
  predicted_labels))
172
 
173
  if not multi_label:
174
+ top_label_key = list(clean_output.keys())[0]
175
+ clean_output = {top_label_key: clean_output[top_label_key]}
176
  return clean_output
177
 
178
  iface = gr.Interface(