AI4Poetry / common /__init__.py
ngartiexauce
Initial clean snapshot (history reset)
45c71f5
"""Common shared code for poetry applications."""
from .model import PoetryModel
from .utils import (
filter_output,
load_prompt_template,
fill_prompt_template,
create_fill_in_blank,
select_words_to_blank,
get_word_definition,
rank_definitions,
load_vocabulary,
load_themes,
load_interests,
check_input_safety,
save_leaderboard_score,
get_leaderboard,
BANNED_WORDS,
UNSAFE_KEYWORDS
)
__all__ = [
'PoetryModel',
'filter_output',
'load_prompt_template',
'fill_prompt_template',
'create_fill_in_blank',
'select_words_to_blank',
'get_word_definition',
'rank_definitions',
'load_vocabulary',
'load_themes',
'load_interests',
'check_input_safety',
'save_leaderboard_score',
'get_leaderboard',
'BANNED_WORDS',
'UNSAFE_KEYWORDS'
]