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

added multi-label as a radio button user input

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -170,7 +170,8 @@ def sequence_to_classify(sequence, labels, multi_label):
170
  sequence,
171
  predicted_labels))
172
 
173
- print("++++clean_output: ", clean_output)
 
174
  return clean_output
175
 
176
  iface = gr.Interface(
@@ -190,3 +191,4 @@ iface = gr.Interface(
190
  examples=prep_examples())
191
 
192
  iface.launch()
 
 
170
  sequence,
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(
 
191
  examples=prep_examples())
192
 
193
  iface.launch()
194
+