.

app.py

@@ -44,10 +44,10 @@ def load_config_from_yaml_file(yaml_path):
 def format_config_display(config):
     if not config:
         return "No configuration loaded"
-    
+
     sections = {
         "Model Architecture": [
-            "hidden_size", "num_layers", "vocab_size", 
+            "hidden_size", "num_layers", "vocab_size",
             "intermediate_size", "tie_word_embeddings", "num_attention_heads", "num_key_value_heads"
         ],
         "Training Configuration": [
@@ -57,14 +57,17 @@ def format_config_display(config):
             "tp", "pp", "dp", "zero_stage"
         ]
     }
-    
-    output = []
-    for section, params in sections.items():
-        output.append(f"\n### {section}")
+
+    output = "<div style='display: flex;'>"
+    for section_name, params in sections.items():
+        output += f"<div style='flex: 1; padding-right: 20px;'><h3>{section_name}</h3>"
         for param in params:
-            output.append(f"- {param}: {config[param]}")
-    
-    return "\n".join(output)
+            value = config.get(param, 'N/A')
+            output += f"<b>{param}</b>: {value}<br>"
+        output += "</div>"
+    output += "</div>"
+    return output
+
 
 def process_yaml_and_plot(config):
     if not config: