kaushikbar
commited on
Commit
·
24bcb3c
1
Parent(s):
1edd175
added multi-label as a radio button user input
Browse files
app.py
CHANGED
@@ -170,7 +170,8 @@ def sequence_to_classify(sequence, labels, multi_label):
|
|
170 |
sequence,
|
171 |
predicted_labels))
|
172 |
|
173 |
-
|
|
|
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 |
+
|