Spaces:
Running
Running
Update utils/target_classifier.py
Browse files
utils/target_classifier.py
CHANGED
|
@@ -37,7 +37,7 @@ def get_target_labels(preds):
|
|
| 37 |
index_of_one = ele.index(1)
|
| 38 |
except ValueError:
|
| 39 |
index_of_one = "NA"
|
| 40 |
-
|
| 41 |
# Retrieve the name of the label (if no prediction made = NA)
|
| 42 |
if index_of_one != "NA":
|
| 43 |
name = label_dict[index_of_one]
|
|
|
|
| 37 |
index_of_one = ele.index(1)
|
| 38 |
except ValueError:
|
| 39 |
index_of_one = "NA"
|
| 40 |
+
st.write(index_of_one)
|
| 41 |
# Retrieve the name of the label (if no prediction made = NA)
|
| 42 |
if index_of_one != "NA":
|
| 43 |
name = label_dict[index_of_one]
|