# Author: Ilya Zelenskiy (proxy3d)
# Telegram channel: https://t.me/greenruff
# Communication Styles LLM: https://iproxy3d.github.io/communication-styles-llm/

from goemotions_en_ru import EmotionClassifier

MODEL_ID = "proxy3d/multi-motions-28"

clf = EmotionClassifier(MODEL_ID)

for text in [
    "I finally finished it, what a relief!",
    "Я наконец закончил — какое облегчение!",
]:
    result = clf.predict(text, top_k=5)
    print(text)
    for item in result["top"]:
        print(f"  {item['label']:14s} {item['score']:.3f}")
    print("active:", result["labels"])