update bytes.

src/streamlit_app.py

@@ -1,9 +1,8 @@
 import os
 import re
 from pathlib import Path
-
-import altair as alt
 import numpy as np
+
 import pandas as pd
 import streamlit as st
 
@@ -76,11 +75,13 @@ else:
         feature_options = sorted(gifs_df["features"].unique().tolist())
         hidden_options = sorted(gifs_df["hidden_dim"].unique().tolist())
 
-        c1, c2 = st.columns(2)
-        with c1:
-            selected_features = st.selectbox("Number of features (nf)", feature_options)
-        with c2:
-            selected_hidden = st.selectbox("Hidden dimension (nh)", hidden_options)
+        left, _ = st.columns([2, 5])  # empty column eats the space
+        with left:
+            c1, c2 = st.columns([1, 1], gap="small")
+            with c1:
+                selected_features = st.selectbox("Number of features (nf)", feature_options)
+            with c2:
+                selected_hidden = st.selectbox("Hidden dimension (nh)", hidden_options)
 
         # Filter + sort by sparsity (instance)
         filtered = gifs_df[
@@ -99,6 +100,9 @@ else:
             items = filtered.to_dict(orient="records")
             for item in items:
                 caption = f"{item['sparsity']} out of {item['sparsity_out_of']}"
-                st.image(item["path"], caption=caption, width="content")
+
+
+                gif_bytes = Path(item["path"]).read_bytes()
+                st.image(gif_bytes, caption=caption)  # should animate
 
 st.divider()