Update chat_a.py

chat_a.py

@@ -1039,22 +1039,23 @@ def override_emotion_if_needed(text):
     return None
     
 def analyze_emotion(text: str):
-    tokenizer = load_tokenizer()
     model = load_sentiment_model()
+    tokenizer = load_tokenizer()
 
-    inputs = tokenizer(
-        text,
-        return_tensors="pt",
-        truncation=True,
-        padding=False,
-        max_length=256,
-    )
+    override = override_emotion_if_needed(text)
+    if override:
+        return override
+
+    inputs = tokenizer(text, return_tensors="pt", truncation=True)
     with torch.no_grad():
-        logits = model(**inputs).logits
-        probs = F.softmax(logits, dim=1)[0].tolist()
+        logits = model(**inputs).logits        # ← model은 '단일 객체'
+        probs = F.softmax(logits, dim=1)[0]
+
     top_indices = torch.topk(probs, k=5).indices.tolist()
     top_emotions = [(klue_emotions[i], float(probs[i]) * 100) for i in top_indices]
-    top_emotion_groups = list(dict.fromkeys([klue_to_general[i] for i in top_indices if probs[i] > 0.05]))
+    top_emotion_groups = list(dict.fromkeys(
+        [klue_to_general[i] for i in top_indices if probs[i] > 0.05]
+    ))
     return top_emotions, top_emotion_groups
 
 def detect_intent(user_input):