Update README.md

README.md

@@ -135,14 +135,12 @@ conversation = f"""<|im_start|>system\nYou are an AI that code. Please answer wi
 
 while True:
     user_input = input("You: ")
-    llm_prompt = f"{conversation}{user_input}<|eot_id|><|start_header_id|>assistant<|end_header_id|>\n\n"
+    llm_prompt = f"{conversation}{user_input}<|im_end|>\n<|im_start|>assistant\nSure! Let me provide a complete and a thorough answer to your question, with functional and production ready code.\n"
     answer = generate_text(llm_prompt)
+    print("=" * 132)
     print(answer)
-
-    conversation = f"{llm_prompt}{answer}<|eot_id|><|start_header_id|>user<|end_header_id|>\n\n"
-
-    json_data = {"prompt": user_input, "answer": answer}
-
-    with open(output_file_path, "a") as output_file:
-        output_file.write(json.dumps(json_data) + "\n")
+    answer_output = extract_output(answer)
+    print("=" * 132)
+    print(answer_output)
+    conversation = f"{llm_prompt}{answer_output}<|im_end|>\n<|im_start|>user\n"
 ```