leavoigt commited on
Commit
46f16fa
1 Parent(s): 10d367b

Update utils/target_classifier.py

Browse files
Files changed (1) hide show
  1. utils/target_classifier.py +1 -1
utils/target_classifier.py CHANGED
@@ -9,7 +9,7 @@ import streamlit as st
9
  from transformers import pipeline
10
 
11
  ## Labels dictionary ###
12
- _lab_dict = {
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
  }