Update README.md

README.md

@@ -1,14 +1,3 @@
----
-license: cc-by-nc-4.0
-language:
-- en
-datasets:
-- Universal-NER/Pile-NER-type
----
-
----
-
-
 # UniNER-7B-all
 
 **Description**: This model is self fine-tuned from [TinyLLama](https://huggingface.co/TinyLlama/TinyLlama-1.1B-intermediate-step-1431k-3T). It is trained on ChatGPT-generated [Pile-NER-type data](https://huggingface.co/datasets/Universal-NER/Pile-NER-type).
@@ -16,16 +5,15 @@ datasets:
 Check our [paper](https://arxiv.org/abs/2308.03279) for more information.
 
 ## Inference
+```python
 # Install transformers from source - only needed for versions <= v4.34
 # pip install git+https://github.com/huggingface/transformers.git
 # pip install accelerate
-
 import torch
 from transformers import pipeline
 
 pipe = pipeline("text-generation", model="/media/4TB_1/minhtt/Documents/test/axolotl/lora-out/merged",
                 torch_dtype=torch.bfloat16, device_map="auto")
-# We use the tokenizer's chat template to format each message - see https://huggingface.co/docs/transformers/main/en/chat_templating
 messages = [
     {
         "role": "system",