Тестирование модели

import spacy
from pprint import pprint


def test_model(input_data: str, path2model: str = '../res/model_emotionsV1'):
    # Загружаем сохраненную модель
    loaded_model = spacy.load(path2model)
    parsed_text = loaded_model(input_data)
    # Определяем возвращаемое предсказание
    prediction = sorted(
        parsed_text.cats.keys(), key=lambda x: parsed_text.cats[x], reverse=True
    )
    score = ', '.join([f"{parsed_text.cats[i]:3f}" for i in prediction])

    # Отладочная инфа
    pprint(parsed_text.cats)
    print(input_data)

    print(
        f"Текст: {input_data}\n\
Предсказание: {', '.join(prediction)}\n\
Score: {score}"
    )


test_model(input())