23 lines
716 B
Python
23 lines
716 B
Python
"""ComfyUI-AgeShift: face age detection + age-conditioned prompt building."""
|
|
|
|
from .nodes.detector_loader import AgeDetectorLoader
|
|
from .nodes.detect import AgeDetect
|
|
from .nodes.prompt_builder import AgePromptBuilder, AgeRangeBatch
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"AgeDetectorLoader": AgeDetectorLoader,
|
|
"AgeDetect": AgeDetect,
|
|
"AgePromptBuilder": AgePromptBuilder,
|
|
"AgeRangeBatch": AgeRangeBatch,
|
|
}
|
|
|
|
NODE_DISPLAY_NAME_MAPPINGS = {
|
|
"AgeDetectorLoader": "🎂 Age Detector Loader",
|
|
"AgeDetect": "🎂 Age Detect",
|
|
"AgePromptBuilder": "🎂 Age Prompt Builder",
|
|
"AgeRangeBatch": "🎂 Age Range Batch",
|
|
}
|
|
|
|
WEB_DIRECTORY = None
|
|
__all__ = ["NODE_CLASS_MAPPINGS", "NODE_DISPLAY_NAME_MAPPINGS"]
|