replace st.secrets

src/model_utils.py

@@ -121,15 +121,15 @@ def save_model(model, user_model_name, metrics_result_single=None):
 
     # Upload to HF dataset
     scheduler = CommitScheduler(
-        repo_id=st.secrets["HF_REPO_ID"],
+        repo_id=os.environ["HF_REPO_ID"],
         repo_type="dataset",
         path_in_repo="models",
-        token=st.secrets["HF_TOKEN"],
+        token=os.environ["HF_TOKEN"],
         private=True,
         folder_path="dummy"
     )
     scheduler.api.upload_file(
-        repo_id=st.secrets["HF_REPO_ID"],
+        repo_id=os.environ["HF_REPO_ID"],
         repo_type="dataset",
         path_in_repo=f"models/{uuid.uuid4()}.parquet",
         path_or_fileobj=buf
@@ -209,15 +209,15 @@ def save_model_ensemble(models, user_model_name, best_iterations=None, fold_scor
     buf.seek(0)
 
     scheduler = CommitScheduler(
-        repo_id=st.secrets["HF_REPO_ID"],
+        repo_id=os.environ["HF_REPO_ID"],
         repo_type="dataset",
         path_in_repo="models",
-        token=st.secrets["HF_TOKEN"],
+        token=os.environ["HF_TOKEN"],
         private=True,
         folder_path="dummy"
     )
     scheduler.api.upload_file(
-        repo_id=st.secrets["HF_REPO_ID"],
+        repo_id=os.environ["HF_REPO_ID"],
         repo_type="dataset",
         path_in_repo=f"models/{uuid.uuid4()}.parquet",
         path_or_fileobj=buf
@@ -246,9 +246,9 @@ def load_model(model_name):
         login(token=os.environ["HF_TOKEN"])
 
     files = hf_hub_download(
-        repo_id=st.secrets["HF_REPO_ID"],
+        repo_id=os.environ["HF_REPO_ID"],
         repo_type="dataset",
-        token=st.secrets["HF_TOKEN"],
+        token=os.environ["HF_TOKEN"],
         filename=None,  # Get whole repo listing
         cache_dir=None,
         local_dir=None,
@@ -258,18 +258,18 @@ def load_model(model_name):
     )
 
     from huggingface_hub import HfApi
-    api = HfApi(token=st.secrets["HF_TOKEN"])
-    all_files = api.list_repo_files(repo_id=st.secrets["HF_REPO_ID"], repo_type="dataset")
+    api = HfApi(token=os.environ["HF_TOKEN"])
+    all_files = api.list_repo_files(repo_id=os.environ["HF_REPO_ID"], repo_type="dataset")
     model_files = [f for f in all_files if f.startswith("models/") and f.endswith(".parquet")]
 
     # Find matching filename
     target_file = None
     for f in model_files:
         downloaded = hf_hub_download(
-            repo_id=st.secrets["HF_REPO_ID"],
+            repo_id=os.environ["HF_REPO_ID"],
             repo_type="dataset",
             filename=f,
-            token=st.secrets["HF_TOKEN"]
+            token=os.environ["HF_TOKEN"]
         )
         table = pq.read_table(downloaded)
         row = table.to_pylist()[0]