Update README.md

README.md

@@ -1,18 +1,51 @@
 # Kat-Gen1
 
+---
+license: apache-2.0
+language:
+- en
+pipeline_tag: text-generation
+tags:
+- text-generation
+- causal-lm
+- pytorch
+- transformers
+library_name: transformers
+datasets:
+- custom
+metrics:
+- perplexity
+- bleu
+- rouge
+base_model: gpt-neox
+---
+
 ## Model Card
 
 | Attribute | Value |
 |-----------|-------|
-| Model Type | Causal Language Model |
-| Architecture | GPT-NeoX |
-| Parameters | ~1.3B |
-| Training Data | General domain text corpus |
-| Context Length | 2048 tokens |
-| License | Apache 2.0 |
-| Language | English |
-| Precision | FP16/FP32 |
-| Framework | PyTorch, Transformers |
+| **Model Name** | Kat-Gen1 |
+| **Model ID** | Katisim/Kat-Gen1 |
+| **Model Type** | Causal Language Model |
+| **Architecture** | GPT-NeoX |
+| **Parameters** | ~1.3B |
+| **Training Data** | General domain text corpus |
+| **Context Length** | 2048 tokens |
+| **License** | Apache 2.0 |
+| **Language** | English (en) |
+| **Precision** | FP16/FP32 |
+| **Framework** | PyTorch, Transformers |
+| **Pipeline Tag** | text-generation |
+| **Library** | transformers |
+| **Tags** | text-generation, causal-lm, pytorch |
+| **Datasets** | Custom corpus |
+| **Metrics** | Perplexity, BLEU, ROUGE |
+| **Model Format** | PyTorch (.bin), SafeTensors |
+| **Tokenizer** | GPT-NeoX BPE |
+| **Vocabulary Size** | 50,304 tokens |
+| **Hidden Size** | 2048 |
+| **Layers** | 24 |
+| **Attention Heads** | 16 |
 
 ## Model Overview