vulnerability_2_1

Running

leavoigt commited on Jan 5

Commit

e5b30ca

•

1 Parent(s): c6ebb8e

Update utils/target_classifier.py

Files changed (1) hide show

utils/target_classifier.py CHANGED Viewed

@@ -6,7 +6,8 @@ from pandas import DataFrame, Series
 from utils.config import getconfig
 from utils.preprocessing import processingpipeline
 import streamlit as st
-from transformers import pipeline
 ## Labels dictionary ###
 label_dict = {

 from utils.config import getconfig
 from utils.preprocessing import processingpipeline
 import streamlit as st
+from setfit import SetFitModel
 ## Labels dictionary ###
 label_dict = {