Omarelrayes commited on
Commit
bc61eae
·
verified ·
1 Parent(s): b6a3870

Create sitecustomize.py

Browse files
Files changed (1) hide show
  1. 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}")