small fixes.

Dockerfile

@@ -6,10 +6,16 @@ RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
     git \
+    git-lfs \
     && rm -rf /var/lib/apt/lists/*
 
-COPY requirements.txt ./
-COPY src/ ./src/
+COPY . .
+
+RUN git lfs install && \
+    git lfs pull && \
+    echo "After LFS pull, showing GIF sizes:" && \
+    (ls -lah /app/src/hf_gifs 2>/dev/null || true) && \
+    (ls -lah /app/src/hf_gifs 2>/dev/null || true)
 
 # Debug: show what got copied
 RUN echo "Contents of /app/src:" && ls -lah /app/src && \

src/streamlit_app.py

@@ -97,8 +97,8 @@ else:
 
             # Render in rows of 4
             items = filtered.to_dict(orient="records")
-            for i in range(0, len(items), 4):
-                cols = st.columns(4)
+            for i in range(0, len(items), 2):
+                cols = st.columns(2)
                 for j, item in enumerate(items[i : i + 4]):
                     caption = f"{item['sparsity']} out of {item['sparsity_out_of']}"
                     cols[j].image(item["path"], caption=caption, use_container_width=True)