adding details about model sizes

app.py

@@ -99,8 +99,7 @@ with gr.Blocks(css=css) as demo:
         """
         # 🚀 POSEIDON: Foundation Models for PDEs 🌊🔬
 
-        POSEIDON is a **foundation model** for solving **Partial Differential Equations (PDEs)** efficiently. Instead of training a separate model for each PDE, POSEIDON **learns a general solution operator**—allowing it to **generalize across different physics** with minimal data.
-        Think of it as the **GPT-4 for PDEs**, trained on a diverse set of **fluid dynamics equations** and capable of adapting to **new, unseen physical systems**. 🌍⚡
+        POSEIDON is a **foundation model** for solving **Partial Differential Equations (PDEs)** efficiently. Instead of training a separate model for each PDE, POSEIDON **learns a general solution operator**—allowing it to **generalize across different physics** with minimal data. Think of it as the **GPT-4 for PDEs**, trained on a diverse set of **fluid dynamics equations** and capable of adapting to **new, unseen physical systems**. 🌍⚡
 
         # **Dataset Explorer**
         POSEIDON provides solutions to a variety of fluid dynamics problems. Below are a few datasets you can explore:
@@ -188,6 +187,9 @@ with gr.Blocks(css=css) as demo:
         ---
 
         ## 🌟 Try POSEIDON Now!
+
+        You can experiment and empower your research with POSEIDON-T (21M parameters), POSEIDON-B (158M parameters), and POSEIDON-L (629M parameters).
+
         🔗 **Pretrained models & datasets**: [Hugging Face Hub](https://huggingface.co/camlab-ethz)  
         📜 **Code & Paper**: [GitHub](https://github.com/camlab-ethz/poseidon) | [arXiv](https://arxiv.org/abs/2405.19101)  
         💬 **Join the Discussion**: [Hugging Face Forums](https://discuss.huggingface.co/)