Upload __init__.py with huggingface_hub

__init__.py

@@ -25,14 +25,13 @@ __email__ = "lea.attia@gmail.com"
 
 # Import main components for easy access
 try:
-    from .color_model import ColorCLIP, Tokenizer
+    from .training.color_model import ColorCLIP
     from .training.hierarchy_model import Model as HierarchyModel, HierarchyExtractor
     from .example_usage import load_models_from_hf, example_search
-    import config
+    from . import config
     
     __all__ = [
         'ColorCLIP',
-        'Tokenizer',
         'HierarchyModel',
         'HierarchyExtractor',
         'load_models_from_hf',