Spaces:
Sleeping
Sleeping
Update utils/target_classifier.py
Browse files
utils/target_classifier.py
CHANGED
|
@@ -116,7 +116,7 @@ def target_classification(haystack_doc:pd.DataFrame,
|
|
| 116 |
# Get predictions
|
| 117 |
predictions = classifier_model(list(haystack_doc.text))
|
| 118 |
st.write("predictions made")
|
| 119 |
-
st.write(predictions
|
| 120 |
# Get labels for predictions
|
| 121 |
pred_labels = get_target_labels(predictions)
|
| 122 |
st.write("pred_labels")
|
|
|
|
| 116 |
# Get predictions
|
| 117 |
predictions = classifier_model(list(haystack_doc.text))
|
| 118 |
st.write("predictions made")
|
| 119 |
+
st.write(predictions)
|
| 120 |
# Get labels for predictions
|
| 121 |
pred_labels = get_target_labels(predictions)
|
| 122 |
st.write("pred_labels")
|