Upload example_usage.py with huggingface_hub

example_usage.py

@@ -37,14 +37,14 @@ def load_models_from_hf(repo_id: str, cache_dir: str = "./models_cache"):
     print("  📦 Loading color model...")
     color_model_path = hf_hub_download(
         repo_id=repo_id,
-        filename="color_model.pt",
+        filename="models/color_model.pt",
         cache_dir=cache_dir
     )
     
     # Loading vocabulary
     vocab_path = hf_hub_download(
         repo_id=repo_id,
-        filename=config.tokeniser_path,
+        filename="tokenizer_vocab.json",
         cache_dir=cache_dir
     )
     
@@ -66,7 +66,7 @@ def load_models_from_hf(repo_id: str, cache_dir: str = "./models_cache"):
     print("  📦 Loading hierarchy model...")
     hierarchy_model_path = hf_hub_download(
         repo_id=repo_id,
-        filename=config.hierarchy_model_path,
+        filename="models/hierarchy_model.pth",
         cache_dir=cache_dir
     )
     
@@ -88,7 +88,7 @@ def load_models_from_hf(repo_id: str, cache_dir: str = "./models_cache"):
     print("  📦 Loading main CLIP model...")
     main_model_path = hf_hub_download(
         repo_id=repo_id,
-        filename=config.main_model_path,
+        filename="models/gap_clip.pth",
         cache_dir=cache_dir
     )