Update README.md

README.md

@@ -7,7 +7,7 @@ language:
 pipeline_tag: text-generation
 ---
 
-# THIS IS 5th PROTOTYPE OF MERAK-7B-v2!
+# Happy to announce the release of our first model, Merak-7B!
 
 Merak-7B is the Large Language Model of Indonesia Languange 
 
@@ -53,7 +53,7 @@ model = AutoModelForCausalLM.from_pretrained(model_id,
 tokenizer = LlamaTokenizer.from_pretrained(model_id)
 
 def generate_response(question: str) -> str:
-  prompt = f"<|prompt|>{question}<|answer|>".strip()
+  prompt = f"<|prompt|>{question}\n<|answer|>".strip()
 
   encoding = tokenizer(prompt, return_tensors='pt').to("cuda")
   with torch.inference_mode():
@@ -93,7 +93,7 @@ model = AutoModelForCausalLM.from_pretrained(model_id,
 tokenizer = LlamaTokenizer.from_pretrained(model_id)
 
 def generate_response(question: str) -> str:
-  prompt = f"<|prompt|>{question}<|answer|>".strip()
+  prompt = f"<|prompt|>{question}\n<|answer|>".strip()
 
   encoding = tokenizer(prompt, return_tensors='pt').to("cuda")
   with torch.inference_mode():