sitecustomize.py

# sitecustomize.py
# 🔥 Automatically registers HF Plugin 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
    
    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}")