Update README.md

README.md

@@ -8,13 +8,12 @@ tags:
 - surrogate
 - heat-equation
 - diffusion
-base_model: thuerey-group/pde-transformer
+base_model: pde-transformer
 ---
 
-# PDE-Transformer mc-s – 2D Diffusion (Heat Equation) Fine-Tuned
+# Diffusion-Transformer mc-s – 2D Diffusion (Heat Equation)
 
-This repository provides a **fine-tuned version of PDE-Transformer (mc-s)** specialized for
-2D heat diffusion on regular grids. The base model is the mixed-channel small variant
+This repository provides a **fine-tuned PDE-Transformer** for better accurate 2D heat diffusion predictions on regular grids. The base model is the mixed-channel small variant
 from `thuerey-group/pde-transformer`, further trained on synthetic solutions of the
 2D diffusion / heat equation with Gaussian bump initial conditions.