Update src/streamlit_app.py

src/streamlit_app.py

@@ -237,7 +237,9 @@ sns.heatmap(
     annot=True,
     fmt="d",
     cmap="YlOrRd",
-    ax=ax
+    ax=ax,
+    annot_kws={"size": 6},               # smaller numbers in cells
+    cbar_kws={"shrink": 0.5}             # shrink the colorbar
 )
     
 # 3. Set titles/labels on the Axes