Update README.md

README.md

@@ -14,7 +14,7 @@ LoRA (QLoRA) training technique.
 
 - **Language:** Romanian
 - **License:** Llama2 Community License Agreement
-- **Finetuned from model:** Llama-2-7b
+- **Finetuned from model:** Meta's Llama2 7B
 
 ### Model Sources
 - **Paper:** https://arxiv.org/abs/2410.04269
@@ -29,13 +29,15 @@ model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, device_map="auto")
 
 input_text = """"Context: În anul 1600, Mihai Viteazul a realizat prima unire a Țărilor Române: Țara Românească, Transilvania și Moldova. Această unire a fost un moment important în istoria României.
 Întrebare: În ce an a realizat Mihai Viteazul prima unire a Țărilor Române?
-Răspuns:"""
+Răspuns: """
 input_ids = tokenizer(input_text, return_tensors="pt")
 
 outputs = model.generate(**input_ids, max_new_tokens=100, eos_token_id = [13])
 print(tokenizer.decode(outputs[0]))
 ```
 
+Note: Adding a space at the end of the prompt has been observed to significantly improve the model's output quality.
+
 ## Training Details and Evaluation
 
 Please refer to the paper for details on the model's training and evaluation.