from transformers import pipeline
import torch

model_id = "Seonghaa/korean-emotion-classifier-roberta"

device = 0 if torch.cuda.is_available() else -1  # GPU 있으면 0, 없으면 CPU(-1)

clf = pipeline(
    "text-classification",
    model=model_id,
    tokenizer=model_id,
    device=device
)

texts = [
    "오늘 길에서 10만원을 주웠어",
    "오늘 친구들이랑 노래방에 갔어",
    "오늘 시험 망쳤어",
]

for t in texts:
    pred = clf(t, truncation=True, max_length=256)[0]
    print(f"입력: {t}")
    print(f"→ 예측 감정: {pred['label']}, 점수: {pred['score']:.4f}
")

출력 예시:

입력: 오늘 길에서 10만원을 주웠어
→ 예측 감정: 기쁨, 점수: 0.9619

입력: 오늘 친구들이랑 노래방에 갔어
→ 예측 감정: 기쁨, 점수: 0.9653

입력: 오늘 시험 망쳤어
→ 예측 감정: 슬픔, 점수: 0.9602