Update README.md

README.md

@@ -45,13 +45,19 @@ The model supports English language.
 
 ## Usage
 
-# CPU
+# CPU and GPU code
 
 ```python
 from transformers import pipeline
+from accelerate import Accelerator
 
-pipe = pipeline("text-generation", model="OEvortex/HelpingAI-Lite")
+# Initialize the accelerator
+accelerator = Accelerator()
 
+# Initialize the pipeline
+pipe = pipeline("text-generation", model="OEvortex/HelpingAI-Lite", device=accelerator.device)
+
+# Define the messages
 messages = [
     {
         "role": "system",
@@ -63,7 +69,13 @@ messages = [
     },
 ]
 
+# Prepare the prompt
 prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
+
+# Generate predictions
 outputs = pipe(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
+
+# Print the generated text
 print(outputs[0]["generated_text"])
+
 ```