minor update

openclimatefix · Jan 10, 2025 · aba04d3 · aba04d3
1 parent 525b9c4
commit aba04d3
Showing 1 changed file with 14 additions and 1 deletion.
diff --git a/src/mlmodel.py b/src/mlmodel.py
@@ -72,7 +72,20 @@ def mlmodel_page():
 
         st.table(all_sites.style.applymap(color_survived, subset=["last_generation_datetime"]))
 
-        # 2. Show site locations on the map
+        # 2. display all models
+        models = get_models(session)
+
+        all_models = pd.DataFrame(
+            [{"name": m.name, "version": m.version, "description": "todo"} for m in models]
+        )
+
+        # order by name
+        all_models = all_models.sort_values(by="name")
+
+        st.write("ML Models")
+        st.write(all_models)
+
+        # 3. Show site locations on the map
         st.subheader("Site Locations on Map")
 
         # Prepare site details for the map