Update README.md

README.md

@@ -46,7 +46,7 @@ HelpingAI-6B has achieved an impressive Emotional Quotient (EQ) of 91.57, making
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
 
-# Load the HelpingAI-3B-coder model
+# Load the HelpingAI-6B model
 model = AutoModelForCausalLM.from_pretrained("OEvortex/HelpingAI-6B", trust_remote_code=True)
 # Load the tokenizer
 tokenizer = AutoTokenizer.from_pretrained("OEvortex/HelpingAI-6B", trust_remote_code=True)
@@ -75,8 +75,6 @@ outputs = model.generate(
     eos_token_id=tokenizer.eos_token_id, 
 )
 
-# Decode the generated text
-output = tokenizer.decode(generated_text[0], skip_special_tokens=True)
 
 response = outputs[0][inputs.shape[-1]:]
 print(tokenizer.decode(response, skip_special_tokens=True))