Update README.md

README.md

@@ -7,6 +7,8 @@ pipeline_tag: text-generation
 library_name: transformers
 tags:
 - text-generation-inference
+base_model:
+- ilsp/Llama-Krikri-8B-Base
 ---
 
 # Llama-Krikri-8B-Instruct: An Instruction-tuned Large Language Model for the Greek language
@@ -51,8 +53,13 @@ tokenizer = AutoTokenizer.from_pretrained("ilsp/Llama-Krikri-8B-Base")
 
 model.to(device)
 
-input_text = tokenizer("Ένα κρικρί διαφέρει απο ένα λάμα επειδή", return_tensors='pt').to(device)
-outputs = model.generate(input_text['input_ids'], max_new_tokens=256, do_sample=True)
+messages = [
+    {"role": "system", "content": "-------"},
+    {"role": "user", "content": "Σε τι διαφέρει ένα κρικρί από ένα λάμα;"},
+]
+prompt = tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False)
+input_prompt = tokenizer(prompt, return_tensors='pt').to(device)
+outputs = model.generate(input_prompt['input_ids'], max_new_tokens=256, do_sample=True)
 
 print(tokenizer.batch_decode(outputs)[0])
 ```