Update model_utils.py

model_utils.py

@@ -52,7 +52,8 @@ def get_model_summary(model_name):
             model = model.to(torch.device("cuda" if torch.cuda.is_available() else "cpu"))
 
         model_summary = str(model) if model else "Model architecture not found."
-        return model_summary, ""
+        config_content = config.to_json_string() if config else "Configuration not found."
+        return f"## Model Architecture\n\n{model_summary}\n\n## Configuration\n\n{config_content}", ""
     except ValueError as ve:
         return "", f"ValueError: {ve}"
     except EnvironmentError as ee: