File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change 6363 jamjam-ai
6464
6565 echo "==== 배포 완료 ===="
66- sudo docker ps | grep jamjam
66+ sudo docker ps | grep jamjam
Original file line number Diff line number Diff line change 22# 로컬 감정분류 모델 로드/추론. 전역 1회 로드로 성능 안정.
33
44import 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 )
1211model = RobertaForSequenceClassification .from_pretrained (MODEL_PATH , local_files_only = True )
1312
1413id2label = {
@@ -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 ]
You can’t perform that action at this time.
0 commit comments