sitecustomize.py

# sitecustomize.py
# 🔥 This file is automatically loaded by Python on startup
# 🔥 It registers the HF Artifact Repository in EVERY Python process

print("🔧 [sitecustomize] Registering HuggingFace Artifact Repository...")

try:
    import sys
    sys.path.insert(0, "/app")
    
    from hf_artifact_repo import HuggingFaceArtifactRepository
    from mlflow.store.artifact import artifact_repository_registry
    
    # Register the plugin
    artifact_repository_registry._artifact_repository_registry.register(
        "hf", HuggingFaceArtifactRepository
    )
    print("✅ [sitecustomize] HuggingFace Artifact Repository registered!")
except Exception as e:
    print(f"⚠️ [sitecustomize] Could not register: {e}")