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

Update utils/target_classifier.py

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