leavoigt commited on
Commit
88a3bb2
·
verified ·
1 Parent(s): fd41e66

Update utils/target_classifier.py

Browse files
Files changed (1) hide show
  1. utils/target_classifier.py +1 -0
utils/target_classifier.py CHANGED
@@ -8,6 +8,7 @@ from utils.preprocessing import processingpipeline
8
  import streamlit as st
9
  from setfit import SetFitModel
10
  from transformers import pipeline
 
11
 
12
  ## Labels dictionary ###
13
  label_dict = {
 
8
  import streamlit as st
9
  from setfit import SetFitModel
10
  from transformers import pipeline
11
+ import os
12
 
13
  ## Labels dictionary ###
14
  label_dict = {