Update app.py

app.py

@@ -39,6 +39,8 @@ def _matrix_payload(np_mat: np.ndarray, labels: Optional[List[str]] = None):
     Return a safe gr.update payload for a Dataframe (headers must match col_count).
     """
     df = pd.DataFrame(np_mat)
+    # Round to 3 decimal places
+    df = df.round(3)
     if labels is not None and len(labels) == df.shape[0]:
         df.index = labels
     if labels is not None and len(labels) == df.shape[1]:
@@ -59,7 +61,7 @@ def _plot_heatmap(D: np.ndarray, labels: Optional[List[str]] = None) -> np.ndarr
     fig, ax = plt.subplots(figsize=(6, 5), dpi=160, facecolor='#1e1e1e')
     ax.set_facecolor('#1e1e1e')
     
-    im = ax.imshow(D, cmap="viridis")
+    im = ax.imshow(D, cmap="magma")
     cbar = plt.colorbar(im, ax=ax, fraction=0.046, pad=0.04)
     cbar.ax.tick_params(colors='white')  # Light colorbar labels