Spaces:
Sleeping
Sleeping
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 |
}
|