Improve model card (#1)

- Improve model card (b33953c2d3b75cfe958fdb2e4cf782a9ac1ae120)


Co-authored-by: Niels Rogge <[email protected]>

README.md

@@ -70,12 +70,15 @@ model-index:
       value: 0.45403922872340424
       stderr: 0.004539171007529716
       verified: false
+library_name: transformers
+pipeline_tag: text-generation
 ---
+
 # Control-LLM-Llama3.1-8B-Math16
 This is a fine-tuned model of Llama-3.1-8B-Instruct for mathematical tasks on OpenMath2 dataset.
 
 ## Linked Paper
-This model is associated with the paper: [Control-LLM](https://arxiv.org/abs/2501.10979).
+This model is associated with the paper: [Control-LLM](https://huggingface.co/papers/2501.10979).
 
 ## Linked Open Source code - training, eval and benchmark
 This model is associated with the github: [Control-LLM](https://github.com/linkedin/ControlLLM).
@@ -112,4 +115,4 @@ The following plot illustrates and compares catastrophic forgetting mitigation d
 ### Alignment Result
 The plot below highlights the alignment result of the model trained with Control LLM.
 
-![Alignment](plots/alignment_best.png)
+![Alignment](plots/alignment_best.png)