Add library_name and pipeline_tag to metadata

README.md

@@ -1,12 +1,15 @@
 ---
-license: apache-2.0
+base_model:
+- hamishivi/tess2_base
 datasets:
 - allenai/tulu-v2-sft-mixture
 language:
 - en
-base_model:
-- hamishivi/tess2_base
+license: apache-2.0
+library_name: transformers
+pipeline_tag: text-generation
 ---
+
 # TESS 2 v0.1 - A Generalist Instruction Tuned Diffusion LM
 
 This model is the instruction-tuned TESS 2. This model is a simplex-based diffusion model adapted from Mistral v0.1 7B, further trained on Dolma 1.7 and Tulu 2 SFT data.
@@ -15,7 +18,6 @@ This is the model based on Mistral v0.1 and Tulu 2.
 
 This model will only work with our custom codebase found [here](https://github.com/hamishivi/tess-2) -- please go there to see details on how to run training and inference.
 
-
 ## Using this model
 
 To run this model, first clone https://github.com/hamishivi/tess-2.
@@ -39,7 +41,6 @@ If you find this work useful, please cite this work as follows.
   year={2025},
   eprint={2502.13917},
   archivePrefix={arXiv},
-  primaryClass={cs.CL},
   url={https://arxiv.org/abs/2502.13917},
  }
 ```