Update README.md

README.md

@@ -61,45 +61,13 @@ The pipeline we used to produce the data and models is fully open-sourced!
 We provide [all instructions](https://nvidia.github.io/NeMo-Skills/openmathreasoning1/)
 to fully reproduce our results, including data generation.
 
-# How to use the models?
+## How to use the models?
 
-Our models can be used in 3 inference modes: chain-of-thought (CoT), tool-integrated reasoning (TIR) and generative solution selection (GenSelect).
-
-To run inference with CoT mode, you can use this example code snippet.
-
-```python
-import transformers
-import torch
-
-model_id = "nvidia/OpenMath-Nemotron-14B-Kaggle"
-
-pipeline = transformers.pipeline(
-    "text-generation",
-    model=model_id,
-    model_kwargs={"torch_dtype": torch.bfloat16},
-    device_map="auto",
-)
-
-messages = [
-    {
-        "role": "user", 
-        "content": "Solve the following math problem. Make sure to put the answer (and only answer) inside \\boxed{}.\n\n" + 
-        "What is the minimum value of $a^2+6a-7$?"},
-]
-
-outputs = pipeline(
-    messages,
-    max_new_tokens=4096,
-)
-print(outputs[0]["generated_text"][-1]['content'])
-```
-
-To run inference with TIR or GenSelect modes, we highly recommend to use our 
+This model will always use code execution to solve math problems, so we highly recommend to run inference with our 
 [reference implementation in NeMo-Skills](https://nvidia.github.io/NeMo-Skills/openmathreasoning1/evaluation/).
 
 Please note that these models have not been instruction tuned on general data and thus might not provide good answers outside of math domain. 
 
-
 ## Citation
 
 If you find our work useful, please consider citing us!
@@ -134,7 +102,7 @@ This model is intended to facilitate research in the area of mathematical reason
 
 Huggingface 04/23/2025 <br> 
 
-## Model Architecture: <br> 
+### Model Architecture: <br> 
 
 **Architecture Type:** Transformer decoder-only language model  <br> 
 
@@ -145,7 +113,7 @@ Huggingface 04/23/2025 <br>
 
 ** This model has 1.5B of model parameters. <br> 
 
-## Input: <br> 
+### Input: <br> 
 
 **Input Type(s):** Text <br> 
 
@@ -157,7 +125,7 @@ Huggingface 04/23/2025 <br>
 
  
 
-## Output: <br> 
+### Output: <br> 
 
 **Output Type(s):** Text <br> 
 
@@ -173,7 +141,7 @@ Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated sys
 
  
 
-## Software Integration : <br> 
+### Software Integration : <br> 
 
 **Runtime Engine(s):** <br> 
 
@@ -195,7 +163,7 @@ Our AI models are designed and/or optimized to run on NVIDIA GPU-accelerated sys
 
  
 
-## Model Version(s): 
+### Model Version(s): 
 
 [OpenMath-Nemotron-1.5B](https://huggingface.co/nvidia/OpenMath-Nemotron-1.5B)