examples/batch.py

# 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)
results = clf.predict_batch([
    "Спасибо, это действительно помогло.",
    "I can't believe this happened again.",
    "Мне тревожно перед завтрашней встречей.",
])

for row in results:
    print(row["text"])
    print(row["top"])