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(
|