Update utils/target_classifier.py
Browse files
utils/target_classifier.py
CHANGED
@@ -9,7 +9,7 @@ import streamlit as st
|
|
9 |
from transformers import pipeline
|
10 |
|
11 |
## Labels dictionary ###
|
12 |
-
|
13 |
'0':'NO',
|
14 |
'1':'YES',
|
15 |
}
|
|
|
9 |
from transformers import pipeline
|
10 |
|
11 |
## Labels dictionary ###
|
12 |
+
label_dict = {
|
13 |
'0':'NO',
|
14 |
'1':'YES',
|
15 |
}
|