Update README.md

README.md

@@ -20,7 +20,9 @@ import transformers
 import torch
 
 model = "eir"
-messages = [{"role": "user", "content": "Vad ska jag göra om jag är förkyld?"}]
+messages = [
+      {"role": "system", "content": "Du är en läkare som är expert i allmänmedicin: Svara på följande fråga:"},
+      {"role": "user", "content": "Vad ska jag göra om jag är förkyld?"}]
 
 tokenizer = AutoTokenizer.from_pretrained(model)
 prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)