Spaces:
Runtime error
Runtime error
Update predict_module.py
Browse files- 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('
|
15 |
-
_, _, le_cat = load('
|
16 |
-
feature_cols = load('
|
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"
|