Update utils/target_classifier.py
Browse files
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
|
|
|
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 = {
|