leavoigt commited on
Commit
ab71d04
1 Parent(s): e5b30ca

Update utils/target_classifier.py

Browse files
Files changed (1) hide show
  1. utils/target_classifier.py +1 -1
utils/target_classifier.py CHANGED
@@ -7,7 +7,7 @@ from utils.config import getconfig
7
  from utils.preprocessing import processingpipeline
8
  import streamlit as st
9
  from setfit import SetFitModel
10
-
11
 
12
  ## Labels dictionary ###
13
  label_dict = {
 
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 = {