Update README.md

README.md

@@ -46,14 +46,31 @@ from transformers import pipeline
 
 # If you want to add a system message, add a dictionary with role "system". However, this will likely have little
 # effect since the model was only finetuned using a single system message.
-messages = [{"role": "user", "content": "Welke talen worden er in België gesproken?"}]
-pipe = pipe = pipeline("text-generation", model="BramVanroy/Llama-2-13b-chat-dutch", device_map="auto")
+messages = [
+    {
+        "role": "user",
+        "content": "Welke talen worden er in België gesproken?"
+    }
+]
+pipe = pipeline(
+    "text-generation",
+    model="BramVanroy/Llama-2-13b-chat-dutch",
+    device_map="auto"
+)
 
 # Just apply the template but leave the tokenization for the pipeline to do
-prompt = pipe.tokenizer.apply_chat_template(messages, tokenize=False)
+prompt = pipe.tokenizer.apply_chat_template(
+    messages,
+    tokenize=False
+)
 
 # Only return the newly generated tokens, not prompt+new_tokens (return_full_text=False)
-generated = pipe(prompt, do_sample=True, max_new_tokens=128, return_full_text=False)
+generated = pipe(
+    prompt,
+    do_sample=True,
+    max_new_tokens=128,
+    return_full_text=False
+)
 
 generated[0]["generated_text"]
 # ' De officiële talen van België zijn Nederlands, Frans en Duits. Daarnaast worden er nog een aantal andere talen gesproken, waaronder Engels, Spaans, Italiaans, Portugees, Turks, Arabisch en veel meer. '