yearye commited on
Commit
9fcfedf
·
verified ·
1 Parent(s): 2f461c2

Update predict_module.py

Browse files
Files changed (1) hide show
  1. predict_module.py +3 -3
predict_module.py CHANGED
@@ -11,9 +11,9 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
11
  from youtube_transcript_api import YouTubeTranscriptApi
12
 
13
  # 모델 및 기타 파일 로드
14
- model = load('model/view_predictor.joblib')
15
- _, _, le_cat = load('model/label_encoders.joblib')
16
- feature_cols = load('model/features.joblib')
17
 
18
  # 감성 분석 모델
19
  senti_model_name = "nlp04/korean_sentiment_analysis_kcelectra"
 
11
  from youtube_transcript_api import YouTubeTranscriptApi
12
 
13
  # 모델 및 기타 파일 로드
14
+ model = load('view_predictor.joblib')
15
+ _, _, le_cat = load('label_encoders.joblib')
16
+ feature_cols = load('features.joblib')
17
 
18
  # 감성 분석 모델
19
  senti_model_name = "nlp04/korean_sentiment_analysis_kcelectra"