update viewer.

src/streamlit_app.py

@@ -5,6 +5,7 @@ import numpy as np
 
 import pandas as pd
 import streamlit as st
+import base64
 
 
 def parse_filename(fname: str):
@@ -104,9 +105,16 @@ else:
             items = filtered.to_dict(orient="records")
             for item in items:
                 caption = f"{item['sparsity']} out of {item['sparsity_out_of']}"
-
-
                 gif_bytes = Path(item["path"]).read_bytes()
-                st.image(gif_bytes, caption=caption)  # should animate
-
+                b64 = base64.b64encode(gif_bytes).decode("ascii")
+
+                st.markdown(
+                    f"""
+                    <figure style="margin:0;">
+                      <img src="data:image/gif;base64,{b64}" style="max-width:100%; height:auto;" />
+                      <figcaption style="font-size:0.9em; color: #666;">{caption}</figcaption>
+                    </figure>
+                    """,
+                    unsafe_allow_html=True,
+                )
 st.divider()