Update README.md

Files changed (1) hide show

README.md CHANGED Viewed

@@ -54,12 +54,12 @@ Laxo Pro is intended for a wide range of applications, including:
 **Example with `transformers.pipeline`:**
 ```python
-# Use a pipeline as a high-level helper
 from transformers import pipeline
 messages = [
     {"role": "user", "content": "Who are you?"},
 ]
-pipe = pipeline("text-generation", model="explorewithai/LaxoPro")
 pipe(messages)
 ```

 **Example with `transformers.pipeline`:**
 ```python
 from transformers import pipeline
 messages = [
+    {"role": "user", "content": "You are alex, a helpful assistant"},
     {"role": "user", "content": "Who are you?"},
 ]
+pipe = pipeline("text-generation", model="explorewithai/LaxoPro", device_map = "cuda")
 pipe(messages)
 ```