Update app.py

app.py

@@ -780,6 +780,8 @@ def train_and_save(
     st.write("Local survival bundle exists:", os.path.exists("survival_bundle.joblib"))
     if os.path.exists("survival_bundle.joblib"):
         st.write("Local survival bundle size (bytes):", os.path.getsize("survival_bundle.joblib"))
+    st.write("Local survival bundle exists:", os.path.exists("survival_bundle.joblib"))
+    st.json(meta.get("survival", {}))
 
     return pipe, meta, X_train, y_test, proba
 
@@ -1011,6 +1013,9 @@ def publish_to_hub(model_repo_id: str, version_tag: str):
             commit_message=f"Update latest survival model ({version_tag})"
         )
 
+    files = api.list_repo_files(repo_id=model_repo_id, repo_type="model")
+
+    st.write([f for f in files if "survival" in f])
 
     return {
         "version_model_path": version_model_path,
@@ -1697,7 +1702,8 @@ with tab_train:
                 st.markdown("**Confusion Matrix (threshold = 0.5)**")
                 st.dataframe(cm_df)
         
-                
+                st.json(meta["survival"])
+