Update README.md

README.md

@@ -23,18 +23,18 @@ This is a base model, it can be finetuned to a particular use case.
 See the snippet below for usage with Transformers:
 
 ```python
->>> import transformers
->>> import torch
+import transformers
+import torch
 
->>> model_id = "AI-Sweden-Models/Llama-3-8B"
+model_id = "AI-Sweden-Models/Llama-3-8B"
 
->>> pipeline = transformers.pipeline(
-    "text-generation",
+pipeline = transformers.pipeline(
+    task="text-generation",
     model=model_id,
     model_kwargs={"torch_dtype": torch.bfloat16},
     device_map="auto"
 )
->>> pipeline(
+pipeline(
     text_inputs="Sommar och sol är det bästa jag vet",
     max_length=128,
     repetition_penalty=1.03)