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"
|