Make display_details robust

app.py

@@ -202,9 +202,7 @@ def display_details(df_1, df_2, sample_idx):
     s_1 = df_1.iloc[sample_idx]
     s_2 = df_2.iloc[sample_idx]
     # Pop model_name and add it to the column name
-    s_1 = s_1.rename(s_1.pop("model_name"))
-    s_2 = s_2.rename(s_2.pop("model_name"))
-    df = pd.concat([s_1, s_2], axis="columns")#.rename_axis("Parameters").reset_index()
+    df = pd.concat([s.rename(s.pop("model_name")) for s in [s_1, s_2] if "model_name" in s], axis="columns")
     return (
         df.style
         .format(na_rep="")