update hidden dimensions.

src/streamlit_app.py

@@ -116,7 +116,7 @@ else:
             # Render in rows of 2
             items = filtered.to_dict(orient="records")
             for item in items:
-                caption = f"{item['sparsity']} out of {item['sparsity_out_of']}"
+                caption = f"{item['sparsity']} out of {item['sparsity_out_of']} for {item['features']} features and {item['hidden_dim']} hidden dimensions"
                 gif_bytes = Path(item["path"]).read_bytes()
                 b64 = base64.b64encode(gif_bytes).decode("ascii")