Update README.md

README.md

@@ -38,7 +38,7 @@ The model follows the Vicuna 1.1 chat format:
 A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions.
 
 USER: {user input}
-ASSISTANT: {machine response}<|endoftext|>
+ASSISTANT:{machine response}<|endoftext|>
 ```
 
 ## Usage
@@ -53,7 +53,7 @@ tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForCausalLM.from_pretrained(model_name)
 
 # Define the prompt in Vicuna 1.1 format
-prompt = "A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions.\n\nUSER: What are the Pyramids of Giza known for?\nASSISTANT: "
+prompt = "A chat between a curious user and an artificial intelligence assistant. The assistant gives helpful, detailed, and polite answers to the user's questions.\n\nUSER: What are the Pyramids of Giza known for?\nASSISTANT:"
 inputs = tokenizer(prompt, return_tensors="pt")
 outputs = model.generate(**inputs, max_length=100)