Skip to content

Commit 1044ea6

Browse files
committed
fix : git action 수정 진행
1 parent 9cfd4e5 commit 1044ea6

2 files changed

Lines changed: 5 additions & 6 deletions

File tree

.github/workflows/deploy.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -63,4 +63,4 @@ jobs:
6363
jamjam-ai
6464
6565
echo "==== 배포 완료 ===="
66-
sudo docker ps | grep jamjam
66+
sudo docker ps | grep jamjam

app/services/emotion_service.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2,13 +2,12 @@
22
# 로컬 감정분류 모델 로드/추론. 전역 1회 로드로 성능 안정.
33

44
import torch
5-
from transformers import RobertaTokenizer, RobertaForSequenceClassification
5+
from transformers import AutoTokenizer, RobertaForSequenceClassification
66

7-
# MODEL_PATH = "outputs_trainer_final2/best_model" # 예: 개발 환경
8-
MODEL_PATH = "/app/best_model" # 예: 컨테이너/서버 환경
7+
MODEL_PATH = "/app/best_model" # 컨테이너/서버 환경
98

109
# 전역 1회 로드
11-
tokenizer = RobertaTokenizer.from_pretrained(MODEL_PATH, local_files_only=True)
10+
tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH, local_files_only=True)
1211
model = RobertaForSequenceClassification.from_pretrained(MODEL_PATH, local_files_only=True)
1312

1413
id2label = {
@@ -26,4 +25,4 @@ def predict_emotion(text: str) -> str:
2625
with torch.no_grad():
2726
outputs = model(**inputs)
2827
pred_id = torch.argmax(outputs.logits, dim=-1).item()
29-
return id2label[pred_id]
28+
return id2label[pred_id]

0 commit comments

Comments
 (0)