Spaces:

TeresaK
/

cpv_test

Runtime error

leavoigt commited on Jan 5

Commit

ab71d04

•

1 Parent(s): e5b30ca

Update utils/target_classifier.py

Files changed (1) hide show

utils/target_classifier.py CHANGED Viewed

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

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