Update 2_hf_repo_info.py

2_hf_repo_info.py

@@ -3,6 +3,12 @@ from huggingface_hub import list_models
 from datetime import date, timedelta, datetime, timezone
 import pandas as pd
 
+# os.environ["KAGGLE_USERNAME"] = "hahunavth"
+# os.environ["KAGGLE_KEY"] = ""
+from kaggle.api.kaggle_api_extended import KaggleApi
+
+api = KaggleApi()
+api.authenticate()
 
 st.set_page_config(layout="wide")
 # @st.cache_resource
@@ -40,6 +46,13 @@ for model in models:
         diff = str(diff)
 
     step = model.id.replace('hahunavth/emofs2-exp', '').split('_')[0]
+
+    kaggle_slug = f"ess-vlsp2023-train-{step}"
+    try:
+        kernel_status = api.kernel_status(user_name="hahunavth", kernel_slug="ess-vlsp2023-train-42")["status"]
+    except:
+        kernel_status = "not found"
+
     _models.append(
         {
             "id": model.id,
@@ -48,6 +61,7 @@ for model in models:
             "huggingface": f'https://huggingface.co/{model.id}/tree/main' if model.id else None,
             "notebook": f"https://www.kaggle.com/code/hahunavth/ess-vlsp2023-train-{step}" if model.id else None,
             "kaggle": f"/kaggle/repo/vlsp2023-ess/config/exp{step}",
+            "status": kernel_status,
         }
     )