Update app.py

Add publish_to_hub uploader

app.py

@@ -6,6 +6,9 @@ import streamlit as st
 import joblib
 import shap
 import matplotlib.pyplot as plt
+import os
+from huggingface_hub import HfApi
+
 
 from sklearn.pipeline import Pipeline
 from sklearn.compose import ColumnTransformer
@@ -173,6 +176,60 @@ def build_shap_explainer(pipe, X_bg, max_bg=200):
     )
     return explainer
 
+def publish_to_hub(model_repo_id: str, version_tag: str):
+    """
+    Uploads model.joblib and meta.json to a Hugging Face *model* repository
+    under a versioned folder and also updates a 'latest/' copy.
+    Requires Space Secret: HF_TOKEN.
+    """
+    token = os.environ.get("HF_TOKEN")
+    if not token:
+        raise RuntimeError("HF_TOKEN not found. Add it in Space Settings → Secrets.")
+
+    api = HfApi(token=token)
+
+    # Versioned paths
+    version_model_path = f"releases/{version_tag}/model.joblib"
+    version_meta_path  = f"releases/{version_tag}/meta.json"
+
+    # Upload versioned artifacts
+    api.upload_file(
+        path_or_fileobj="model.joblib",
+        path_in_repo=version_model_path,
+        repo_id=model_repo_id,
+        repo_type="model",
+        commit_message=f"Upload model artifacts ({version_tag})"
+    )
+    api.upload_file(
+        path_or_fileobj="meta.json",
+        path_in_repo=version_meta_path,
+        repo_id=model_repo_id,
+        repo_type="model",
+        commit_message=f"Upload metadata ({version_tag})"
+    )
+
+    # Also maintain a 'latest/' copy for easy loading
+    api.upload_file(
+        path_or_fileobj="model.joblib",
+        path_in_repo="latest/model.joblib",
+        repo_id=model_repo_id,
+        repo_type="model",
+        commit_message=f"Update latest model ({version_tag})"
+    )
+    api.upload_file(
+        path_or_fileobj="meta.json",
+        path_in_repo="latest/meta.json",
+        repo_id=model_repo_id,
+        repo_type="model",
+        commit_message=f"Update latest metadata ({version_tag})"
+    )
+
+    return {
+        "version_model_path": version_model_path,
+        "version_meta_path": version_meta_path,
+        "latest_model_path": "latest/model.joblib",
+        "latest_meta_path": "latest/meta.json",
+    }
 
 # ============================================================
 # Streamlit UI