Update 2_hf_repo_info.py

2_hf_repo_info.py

@@ -10,6 +10,7 @@ from kaggle.api.kaggle_api_extended import KaggleApi
 
 st.set_page_config(layout="wide")
 
+st.header("Experiments")
 
 @st.cache_resource
 def init_kaggle():
@@ -28,11 +29,11 @@ def get_ckpt_list(model):
     file_names = [file.rfilename for file in files if "pytorch_model" in file.rfilename]
     return file_names
 
-
-with st.form("my_form"):
-    _author = st.text_input("Author", "hahunavth")
-    _filter = st.text_input("Slug", "emofs2")
-    submit = st.form_submit_button("Submit")
+with st.expander("Filter:"):
+    with st.form("my_form"):
+        _author = st.text_input("Author", "hahunavth")
+        _filter = st.text_input("Slug", "emofs2")
+        submit = st.form_submit_button("Submit")
 
 models = list(
     list_models(author=_author, filter=_filter, sort="last_modified", direction=-1)
@@ -87,7 +88,12 @@ async def set_df_kernel_status(df, n=12):
         df.at[i, "status"] = status
         print(i, status)
 
-st.button("Update kernel status", on_click=lambda: get_kernel_status.clear())
+def on_update_kernel_status_click():
+    st.session_state["last_update"] = datetime.now().strftime('%Y-%m-%d %H:%M:%S')
+    get_kernel_status.clear()
+
+st.button("Update kernel status", on_click=on_update_kernel_status_click)
+st.text(f"Last updated: {st.session_state.get('last_update', 'not yet')}")
 
 asyncio.run(set_df_kernel_status(df))