Update README.md

README.md

@@ -47,7 +47,7 @@ model.generate(
     streamer=streamer
 )
 ```
-Classfication section undertraining
+Classfication
 ```py
 import os
 import json
@@ -55,7 +55,7 @@ import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
 # Path to saved model (change if needed)
-ckpt = "HelpingAI/hai3.1-checkpoint-0002"  # or the HF name like HelpingAI/hai3.1-checkpoint-0001
+ckpt = "HelpingAI/hai3.1-checkpoint-0002"
 device = "cpu"
 
 print("Device:", device)
@@ -104,7 +104,6 @@ print("Predicted label:", id2label[pred_id] if pred_id < len(id2label) else "unk
 
 TTS layers in training
 
-NOTE: we have used qwen2 tokenizer in it
 
 This model contains layers from our diffrent models
-To aline layers we have done post training after merging layers 
+To aline layers we have done post-training after merging layers