Spaces:
Sleeping
Sleeping
Create sitecustomize.py
Browse files- sitecustomize.py +18 -0
sitecustomize.py
ADDED
|
@@ -0,0 +1,18 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# sitecustomize.py
|
| 2 |
+
# 🔥 Automatically registers HF Plugin in every Python process
|
| 3 |
+
|
| 4 |
+
print("🔧 [sitecustomize] Registering HuggingFace Artifact Repository...")
|
| 5 |
+
|
| 6 |
+
try:
|
| 7 |
+
import sys
|
| 8 |
+
sys.path.insert(0, "/app")
|
| 9 |
+
|
| 10 |
+
from hf_artifact_repo import HuggingFaceArtifactRepository
|
| 11 |
+
from mlflow.store.artifact import artifact_repository_registry
|
| 12 |
+
|
| 13 |
+
artifact_repository_registry._artifact_repository_registry.register(
|
| 14 |
+
"hf", HuggingFaceArtifactRepository
|
| 15 |
+
)
|
| 16 |
+
print("✅ [sitecustomize] HuggingFace Artifact Repository registered!")
|
| 17 |
+
except Exception as e:
|
| 18 |
+
print(f"⚠️ [sitecustomize] Could not register: {e}")
|