kaushikbar
commited on
Commit
·
d88be3e
1
Parent(s):
24bcb3c
added multi-label as a radio button user input
Browse files
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 |
-
|
|
|
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(
|